BZOJ4337: BJOI2015 树的同构

题解

#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=55,pow=2333,tt=1000000007;
int m,n,tot,s[maxn],lnk[maxn],son[2*maxn],nxt[2*maxn];
long long ans[maxn][maxn];
bool vis[maxn];
void add(int x,int y){
tot++; son[tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;
}
long long dfs(int x){
long long w=pow; int tail=0;
long long a[maxn];
for (int j=lnk[x];j;j=nxt[j]) if (vis[son[j]]==0) vis[son[j]]=1,a[++tail]=dfs(son[j]);
sort(a+1,a+1+tail); for (int i=1;i<=tail;i++) w=((w*pow)%tt+a[i])%tt;
return w;
}
void get_ans(int k,int n){
for (int i=1;i<=k;i++) {
bool check=0;
for (int j=1;j<=n;j++) if (ans[k][j]!=ans[i][j]) check=1;
if (check==0) {printf("%d\n",i); return;}
}
}
int main(){
scanf("%d",&m);
for (int k=1;k<=m;k++) {
scanf("%d",&n); s[k]=n;
tot=0; memset(lnk,0,sizeof(lnk)); memset(nxt,0,sizeof(nxt));
for (int i=1;i<=n;i++) {int x; scanf("%d",&x); if (x==0) continue; add(x,i); add(i,x);}
for (int i=1;i<=n;i++) {
memset(vis,0,sizeof(vis)); vis[i]=1;
ans[k][i]=dfs(i);
}
sort(ans[k]+1,ans[k]+1+n);
get_ans(k,n);
}
return 0;
}