BZOJ4337 BJOI2015 树的同构

　　构造一个靠谱点的树哈希函数，每个树以每个点都为根算一次哈希值，然后归类一下有相同哈希值的树就行了。

　　代码

#include<cstdio>
#include<algorithm>
#define nc() getchar()
#define N 500100
int dp,pre[N],p[N],ans[N],tt[N],n,i,m[N],a,j,k;
long long q,hash[N],s[N],v[N];
void link(int x,int y)
{
    dp++;pre[dp]=p[x];p[x]=dp;tt[dp]=y;
}
void root(int x,int fa)
{
    int i=p[x];
    long long ans=1;
    s[x]=1;
    while (i)
    {
        if (tt[i]!=fa)
        {
            root(tt[i],x);
            long long tmp=v[tt[i]]+131;
            s[x]=s[x]+s[tt[i]];
            v[x]=v[x]+tmp;
        }
        i=pre[i];
    }
    v[x]=v[x]*s[x]%1000000007;
}
int main()
{
    scanf("%d",&n);
    for (i=1;i<=n;i++)
    {
        scanf("%d",&m[i]);
        dp=0;
        for (j=1;j<=m[i];j++) p[j]=0;
        for (j=1;j<=m[i];j++)
        {
            scanf("%d",&a);
            if (a)
            {
                link(a,j);
                link(j,a);
            }
        }
        
        for (j=1;j<=m[i];j++)
        {
            for (k=1;k<=m[i];k++)
            s[k]=0,v[k]=0;
            root(j,0);q=v[j];
            for (k=1;k<i;k++)
            if ((m[k]==m[i])&&(hash[k]==q)) 
            {
                ans[i]=k;break;
            }
        }
        if (ans[i]==0) ans[i]=i,hash[i]=q;
        
    }
    for (i=1;i<=n;i++)
    printf("%d
",ans[i]);
}