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;
}