BZOJ4337:[BJOI2015]树的同构(树hash)

Description

树是一种很常见的数据结构。

我们把N个点，N-1条边的连通无向图称为树。

若将某个点作为根，从根开始遍历，则其它的点都有一个前驱，这个树就成为有根树。

对于两个树T1和T2，如果能够把树T1的所有点重新标号，使得树T1和树T2完全相

同，那么这两个树是同构的。也就是说，它们具有相同的形态。

现在，给你M个有根树，请你把它们按同构关系分成若干个等价类。

Input

第一行，一个整数M。

接下来M行，每行包含若干个整数，表示一个树。第一个整数N表示点数。接下来N

个整数，依次表示编号为1到N的每个点的父亲结点的编号。根节点父亲结点编号为0。

Output

输出M行，每行一个整数，表示与每个树同构的树的最小编号。

Sample Input

4
4 0 1 1 2
4 2 0 2 3
4 0 1 1 1
4 0 1 2 3

Sample Output

1
1
3
1

HINT

【样例解释】

编号为1, 2, 4 的树是同构的。编号为3 的树只与它自身同构。

100% 的数据中，1 ≤ N, M ≤ 50。

Solution

树hash，x的子树计算方法为$\sum hash[i]*val[i]$，其中hash[i]表示的是x的儿子的第i大的哈希值，val[i]是随机的一组很大的数。

Code

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<map>
 #include<algorithm>
 #define LL long long
 #define MOD (998244353)
 using namespace std;

 struct Edge{int to,next;}edge[];
 LL T,n,x,ans,hash[],val[];
 int head[],num_edge;
 map<LL,LL>Map;

 void add(int u,int v)
 {
     edge[++num_edge].to=v;
     edge[num_edge].next=head[u];
     head[u]=num_edge;
 }

 void Dfs(int x,int fa)
 {
     LL q[],tot=;
     hash[x]=;
     for (int i=head[x]; i; i=edge[i].next)
         if (edge[i].to!=fa)
             Dfs(edge[i].to,x),q[++tot]=hash[edge[i].to];
     if (tot==){hash[x]=; return;}
     sort(q+,q+tot+);
     for (int i=; i<=tot; ++i)
         hash[x]=(hash[x]+q[i]*val[i])%MOD;
 } 

 int main()
 {
     for (int i=; i<=; ++i)
         val[i]=rand()*233473ll+rand()*19260817ll+rand();
     scanf("%d",&T);
     for (int t=; t<=T; ++t)
     {
         scanf("%d",&n);
         memset(head,,sizeof(head)); num_edge=;
         for (int i=; i<=n; ++i)
         {
             scanf("%d",&x);
             if (!x) continue;
             add(x,i), add(i,x);
         }
         ans=;
         for (int i=; i<=n; ++i)
         {
             Dfs(i,-);
             if (!Map[hash[i]]) Map[hash[i]]=t;
             else Map[hash[i]]=min(Map[hash[i]],(LL)t);
             ans=min(ans,Map[hash[i]]);
         }
         printf("%lld\n",ans);
     }
 }