zoj 3602 Count the Trees 树的同构

/*
二叉树的同构 
把每棵子树映射为一个整数
如节点n的左右孩子为L,R，那么取子树L对应的整数Ln
子树R对应的整数Rn,把pair(Ln,Rn)对应为一个整数就是
当前子树的映射
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<set>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=100009*4;
int n[2];
struct Node
{
    int l,r,f;//左、右、父
}node[2][maxn];
int deg[maxn],hash[2][maxn],num[2][maxn];
//deg表示节点有多少棵子树
//hash[i][j]表示第i棵树中，以节点j为根的子树对应的哈希值
//num[i][j]表示第i棵树中，哈希值为j的子树有多少棵
long long ans;
map<pair<int,int>,int> my;
void init()
{
    scanf("%d%d",&n[0],&n[1]);
    for(int j=0;j<2;j++)
    for(int i=1;i<=n[j];i++)
    {
        scanf("%d%d",&node[j][i].l,&node[j][i].r);
        if(node[j][i].l!=-1)
        node[j][node[j][i].l].f=i;
        if(node[j][i].r!=-1)
        node[j][node[j][i].r].f=i;
    }
}
void bfs(int t)
{
    int now,L,R;
    queue<int> q;
    deg[0]=3;
    for(int i=1;i<=n[t];i++)
    {
        deg[i]=(node[t][i].l>-1)+(node[t][i].r>-1);
        if(deg[i]==0)
        {
            hash[t][i]=0;
            q.push(i);
        }
    }
    while(!q.empty())
    {

        now=q.front();q.pop();
        L=-1;//左子树的hash值
        if(node[t][now].l>-1)
        L=hash[t][node[t][now].l];
        R=-1;//右子树的hash值
        if(node[t][now].r>-1)
        R=hash[t][node[t][now].r];
        pair<int,int> pi=make_pair(L,R);//当前子树的特征值
        if(my[pi]==0) my[pi]=my.size()+1;//my[pi]当前子树的哈希值
        num[t][my[pi]]++;
        hash[t][now]=my[pi];
        if(t)ans+=num[!t][my[pi]];
        //如果的第二棵树，加上第一棵树中hash值和当前子树hash值相同的子树的个数
        deg[node[t][now].f]--;
        if(deg[node[t][now].f]==0)
        q.push(node[t][now].f);
    }
}
void solve()
{
    my.clear();
    node[0][1].f=node[1][1].f=0;
    memset(num,0,sizeof(num));
    ans=0;
    bfs(0);
    bfs(1);
    cout<<ans<<endl;
}
int main()
{
    int ca;
    scanf("%d",&ca);
    while(ca--)
    {
        init();
        solve();
    }
    return 0;
}