[BJOI2015]树的同构

[BJOI2015]

• 判断两棵无根树是否同构
• 树Hash的原理就是不要让节点编号去影响Hash值
• 那么我们考虑一下用子树大小结合Hash的进制规则来做

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define ull unsigned long long
using namespace std;
const int N=1e3+10;
const ull p=233333;
struct node{int y,n;}e[N];
ull val[N][N],h[N];
int size[N],tot[N],lin[N],id[N],len,m,n,fa;
void read(int x,int y){e[++len].y=y,e[len].n=lin[x],lin[x]=len;}

void dfs(int x,int f){
    tot[x]=1,h[x]=0;
    for(int i=lin[x];i;i=e[i].n){
        int y=e[i].y;
        if(y==f)continue;
        dfs(y,x);
        tot[x]+=tot[y];
        h[x]^=h[y]*p+tot[y];
    }
}
int main()
{
    //freopen("a.in","r",stdin);
    scanf("%d",&m);
    rep(i,1,m){
        memset(lin,0,sizeof(lin));len=0;
        scanf("%d",&size[i]);
        rep(j,1,size[i]){
            scanf("%d",&fa);
            if(fa)read(fa,j),read(j,fa);
        }
        rep(j,1,size[i]){
            dfs(j,0);
            val[i][j]=h[j];
        }
        sort(val[i]+1,val[i]+size[i]+1);
    }
    rep(i,1,m){
        rep(j,1,m){
            if(size[i]==size[j]){
                int flag=1;
                rep(k,1,size[i])if(val[i][k]!=val[j][k]){flag=0;break;}
                if(flag){printf("%d\n",j);break;}
            }
        }
    }return 0;
}