题目梗概
对于两个树T1和T2,如果能够把树T1的所有点重新标号,使得树T1和树T2完全相
同,那么这两个树是同构的。
有M个有根树,请你把它们按同构关系分成若干个等价类。
解题思路
第一次写树hash。
对于一棵树,先把所有子树hash一下,然后在合并成当前树的hash值。
考虑如何合并。
因为我们不能受遍历顺序的影响,所以应该先排序。
然后对于每个子树的hash值再分别乘p,显然这个p不能相同,所以提前构造素数表。
对于这道题,显然要对每个节点为根的状态都hash一下。
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=55,p[]={0,53,59,61,67,71,73,79,83,89,97,101,103,107,109,113,127,131,137,139,149,151,157,163,167,173,179,181,191,193,197,199,211,223,227,229,233,239,241,251,257,263,269,271,277,281,283,293,307,311,313,317};
LL hash[maxn][maxn],sum[maxn];
int T,n,tot,lnk[maxn],son[maxn*2],nxt[maxn*2];
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
void DFS(int x,int fa){
LL s[maxn];int top=0;s[++top]=1;
for (int j=lnk[x];j;j=nxt[j])
if (son[j]!=fa) DFS(son[j],x),s[++top]=sum[son[j]];
sort(s+1,s+1+top);sum[x]=0;
for (int i=1;i<=top;i++) sum[x]+=s[i]*p[i];
}
int main(){
freopen("exam.in","r",stdin);
freopen("exam.out","w",stdout);
scanf("%d",&T);
for (int w=1;w<=T;w++){
scanf("%d",&n);tot=0;
memset(lnk,0,sizeof(lnk));
for (int i=1;i<=n;i++){
int x;scanf("%d",&x);
if (x) add(i,x);add(x,i);
}
for (int i=1;i<=n;i++) DFS(i,0),hash[w][i]=sum[i];
sort(hash[w]+1,hash[w]+1+n);
for (int i=1;i<=w;i++){
int pd=0;
for (int j=1;j<=n;j++)
if (hash[i][j]!=hash[w][j]){pd=1;break;}
if (pd) continue;
printf("%d\n",i);break;
}
}
return 0;
}