E. Preorder（树哈希）

荼白777

已于 2022-04-30 15:33:16 修改

阅读量707

点赞数

分类专栏：哈希文章标签：哈希算法深度优先算法

于 2022-04-30 15:30:11 首次发布

本文链接：https://blog.csdn.net/weixin_45724872/article/details/124513506

版权

树哈希子树本质不同哈希函数 C++

关键词由CSDN通过智能技术生成

哈希专栏收录该内容

1 篇文章 0 订阅

订阅专栏

传送门

题意

在这里插入图片描述

思路

我对哈希一无所知，因此直接贴大佬的题解。
在这里插入图片描述

所以问题主要在于如何确定两颗树是否本质不同；

这里使用树哈希，哈希函数如下；

在这里插入图片描述

当前节点的值为A，那么就取 $P_0$ ；为B，则取 $P_1$

其中 $P_0,P_1$ 为两个大的素数，depth为深度；

这样的哈希函数保证了，本质相同的两棵子树，其哈希值一定也相同。不过严格来说，哈希值相同并不能反过来推定两棵子树是本质相同的，所以该方法还是有可能出错的。

Code

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = (1<<19) + 10;
const int MOD = 998244353;
char str[N];
int n;
ll f[N];
ll h[N];//树哈希
int qpow(int x,int y){
    ll ret = 1,base = x;
    while(y){
        if(y&1) ret = ret * base % MOD;
        base = base * base % MOD;
        y >>= 1;
    }
    return ret % MOD;
}
const int p0 = 1015159,p1 = 1e9+7;
void dfs(int u,int dep){
    if(u*2 > (1<<n)-1){//叶子节点
        f[u] = 1;
        int p = str[u] == 'A'?p0:p1;
        h[u] = qpow(p,dep);
        return;
    }
    int lc = u << 1;
    int rc = u << 1 | 1;
    dfs(lc,dep+1);
    dfs(rc,dep+1);
    int p = str[u] == 'A'?p0:p1;
    h[u] = ((h[lc]*h[rc])%MOD+qpow(p,dep))%MOD;
    f[u] = (f[lc]*f[rc])%MOD;
    //如果左右树不同，再*个2
    if(h[lc] != h[rc]) f[u] = (2*f[u])%MOD;
}
void solve(){
    cin >> n;
    cin >> (str+1);
    dfs(1,1);
    cout << f[1] << '\n';
}
signed main(){
    std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
    return 0;
}

参考资料

题解1
题解2