Aizu - 2784Similarity of Subtrees（树哈希）

题意：

给你一棵树，问有多少对相似的子树（每层深度拥有相同数量的节点）。

思路：

将每棵树通过hash算法映射出一个值，下面这个公式用到了BKDRhash算法

Hash（T）= （ n1+n2P+n3P2+…+nmPm−1 ） % MOD /ni表示当前子树深度为i的节点有几个。

相当于将一棵树转化成一个p进制的整数。

代码:

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int maxn = 1e5+7;
const ll p = 137;
const ll mod = 1e9+7;

int n;
ll Hash[maxn];
vector<int> arr[maxn];
map<ll,ll> mp;
map<ll,ll>::iterator it;

void dfs(int u)
{
    Hash[u] = 1;
    int len = arr[u].size();
    for(int i = 0;i<len;i++)
    {
        int v = arr[u][i];
        dfs(v);
        Hash[u] = (Hash[u]+Hash[v]*p)%mod;
    }
    mp[Hash[u]]++;
}

int main()
{
    scanf("%d",&n);
    for(int i = 0;i<n-1;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        arr[u].push_back(v);
    }
    dfs(1);
    ll ans = 0;
    for(it = mp.begin();it!=mp.end();it++)
    {
        ans += (it->second-1)*it->second/2;
    }
    printf("%lld\n",ans);
    return 0;
}