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;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值