JAG Practice2016-E Similarity of Subtrees- hash

https://acm.bnu.edu.cn/v3/statments/jag2016.pdf

题意：

num(T1，d)  为树T1在深度为d这一层的子节点数量

定义 两棵树相似 的条件为

对所有的d， S(T,d) equals S(T ′ ,d) 

从叶子节点开始bfs，每个节点的相似度信息可以hash成一个数。

从当前节点开始去更新父亲节点的相似度信息

只有当一个节点的所有儿子都访问完了（也即得到了自身完整的相似度信息），才进入队列

在每一层计算一次，具有相同相似度的节点对答案的贡献为 C(X,2)

#include <cstdio>#include <cmath>#include <cstring>#include <string>#include <algorithm>#include <iostream>#include <queue>#include <map>#include <set>#include <vector>using namespace std;const int N=100005;typedef long long ll;vector<int >mp[N];int son_num[N];int son_come[N];unsigned long long  hash_val[N];int fa[N];void dfs(int x,int f){    for (int i=0; i<mp[x].size(); i++)    {        int v=mp[x][i];        if (v==f)continue;        fa[v]=x;        son_num[x]++;        dfs(v,x);    }}map<unsigned long long   ,int> vis;map<unsigned long long  ,int> ::iterator it;struct node{    int dep,x;    node() {}    node(int a,int b)    {        x=a,dep=b;    }};queue<node> q;int main(){    int n,u,v;    cin>>n;    for (int i=1; i<n; i++)    {        scanf("%d%d",&u,&v);        mp[u].push_back(v);        mp[v].push_back(u);    }    dfs(1,0);    for (int i=1; i<=n; i++)        if (!son_num[i]) q.push(node(i,1));    ll ans=0;    while(!q.empty())    {        int sz=q.size();        vis.clear();        for (int i=1; i<=sz; i++)        {            node tp=q.front();            q.pop();            hash_val[tp.x]+=239;            vis[hash_val[tp.x]]++;            int f=fa[tp.x];            hash_val[f]+=hash_val[tp.x]*239;            son_come[f]++;            if (son_come[f]==son_num[f])                q.push(node(f,tp.dep+1));        }        for (it=vis.begin();it!=vis.end();it++)        ans+= 1LL*it->second*(it->second-1)/2;    }    printf("%lld\n",ans);    return 0;}