USTC 1117

07年的论文,hash的应用。

用hash可以在nlogn的时间内解决有根树的同构判定和无根树的同构判定。

由于没找到靠谱的无根树同构模板题(无根树的同构比较麻烦,写个题确认一下)

于是看到了这道题,第一次知道中科大还有OJ(⊙﹏⊙)

而且貌似不能用TAT

我也不知道Restricted Function是什么鬼,反正就是评测不了,不过把样例过了倒是。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
typedef unsigned long long ll;
struct Edge{int to,next;}e[2005];
int head[1005],cnt;
void ins(int u,int v){
	e[++cnt]=(Edge){v,head[u]};head[u]=cnt;
}
const int p=233;
int xp[1005],son[1005];
vector<ll>v[1005];
ll w[1005];
ll calc(int u){
	ll ans=0;
	for(int i=0;i<v[u].size();i++)
	ans=ans*p+v[u][i];
	return ans+xp[son[u]];
}
ll dfs(int u,int fa){
	son[u]=0;
	ll ans=0;
	v[u].clear();
	for(int i=head[u];i;i=e[i].next)
	if(e[i].to!=fa){
		v[u].push_back(dfs(e[i].to,u));
		son[u]++;
	}
	sort(v[u].begin(),v[u].end());
	return w[u]=calc(u);
}
int find(int u,ll x){
	int l=0,r=v[u].size()-1;
	while(l<=r){
		int mid=l+r>>1;
		if(v[u][mid]<=x)l=mid+1;
		else r=mid-1;
	}
	return l-1;
}
void dfs1(int u,int fa,vector<ll>&g){
	int n=v[u].size();
	vector<ll>f1,f2;
	f1.resize(n);f2.resize(n);
	ll ans=0;
	sort(v[u].begin(),v[u].end());
	for(int i=0;i<n;i++)
	f1[i]=ans=ans*p+v[u][i];
	ans=0;
	for(int i=n-1;i>=0;i--)
	f2[i]=ans=ans+v[u][i]*xp[n-1-i];
	for(int i=head[u];i;i=e[i].next){
		int t=e[i].to;if(t==fa)continue;
		int k=find(u,w[t]);
		ans=0;
		if(k>0)ans+=f1[k-1]*xp[n-1-k];
		if(k<n-1)ans+=f2[k+1];
		v[t].push_back(ans+xp[son[u]-(fa<0?1:0)]);
		son[t]++;
		g.push_back(calc(t));
		son[t]--;
		dfs1(t,u,g);
	}
}
void solve(int n,vector<ll>& f){
	int u,v;
	memset(head,0,sizeof(head));cnt=0;
	for(int i=1;i<n;i++){
		scanf("%d%d",&u,&v);
		ins(u,v);ins(v,u);
	}
	f.clear();
	f.push_back(dfs(1,-1));
	dfs1(1,-1,f);
	sort(f.begin(),f.end());
}
vector<ll>f,g;
int main(){
	int T;scanf("%d",&T);
	xp[0]=1;
	for(int i=1;i<=1000;i++)xp[i]=xp[i-1]*p;
	while(T--){
		int n;scanf("%d",&n);
		solve(n,f);solve(n,g);
		puts(f==g?"same":"different");
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值