bzoj4337树的同构

树的同构

采用hash的方法判断是否同构
把每棵树的每个节点当作根节点求一次hash
我的计算方法:
将一个节点的儿子的hash排序
$Hash[u]=\sum i*hash[i]*prime[i]*(i|prime[i])*(i \, xor \, prime[i])*hash[i])mod \,1000000007$
不要问我问什么这么变态
WA了之后乱改的
贴代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cctype>
using namespace std;
const int mod=1000000007;
inline void read(int &x)
{
    int s=0,w=1;
    char c=getchar();
    while(!isdigit(c)){if(c=='-')w=-1;c=getchar();}
    while(isdigit(c)){s=(s<<3)+(s<<1)+c-'0',c=getchar();}
    x=s*w;
}
const int MAXN=1010;
const int N=60;
struct edge
{
    int to,nxxt;
}e[N<<1];
int prime[MAXN],tot,head[N],cnt,Hash[N][N],_hash[N],root,m,ans[N],sum[N];
inline void add(int x,int y)
{
    e[++cnt].to=y;
    e[cnt].nxxt=head[x];
    head[x]=cnt;
}
inline void clear()
{
    memset(head,0,sizeof(head));
    cnt=0;
    memset(_hash,0,sizeof(_hash));
}
bool not_prime[MAXN];
inline void init()
{
    for(register int i=2;i<=1000;i++)
    {
        if(!not_prime[i])prime[++tot]=i;
        for(register int j=1;j<=tot&&prime[j]*i<=1000;j++)
        {
            not_prime[i*prime[j]]=true;
            if(i%prime[j]==0)break;
        }
    }
}
inline void dfs(int u,int fa)
{
    int temp[N],tempcnt=0;
    temp[++tempcnt]=1;
    for(register int i=head[u];i;i=e[i].nxxt)
    {
        if(e[i].to==fa)continue ;
        dfs(e[i].to,u);
        temp[++tempcnt]=_hash[e[i].to];
    }
    sort(temp+1,temp+tempcnt+1);
    for(register int i=1;i<=tempcnt;i++)
        _hash[u]=(_hash[u]+(i^temp[i])*prime[i]*(i|prime[i])*(i^prime[i])*temp[i])%mod;
}
int main()
{
    init();
    read(m);
    for(register int i=1;i<=m;i++)
    {
        clear();
        read(sum[i]);
        for(register int j=1;j<=sum[i];j++)
        {
            int fa;
            read(fa);
            if(!fa)continue ;
            add(fa,j),add(j,fa);
        }
        for(register int j=1;j<=sum[i];j++)
        {
            memset(_hash,0,sizeof(_hash));
            dfs(j,0);
            Hash[i][j]=_hash[j];
        }
    }
    for(register int i=1;i<=m;i++)
    {
        ans[i]=i;
        for(register int j=1;j<i;j++)
        {
            if(sum[i]^sum[j])continue ;
            bool flag=false;
            for(register int k=1;k<=sum[i];k++)
            {
                for(register int p=1;p<=sum[j];p++)
                {
                    if(Hash[i][k]==Hash[j][p])
                    {
                        ans[i]=j;
                        flag=true;
                        break;
                    }
                }
                if(flag)break;
            }
            if(flag)break;
        }
        printf("%d\n",ans[i]);
    }
}