ZOJ 3602 Count the Trees 树的同构 （哈希）

链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4705

题意：给出两棵二叉树A和B，问分别处于A中的子树a和处于B中的子树b结构相同的有多少对。

思路：哈希的想法，不同的数字对应的是不同的结构，比如1代表着单独的叶子结点，2代表着有左子树是叶子结点而没有右子树的子树...每出现一种新的子树情形就记录下来，记录的方式是用dfs回溯过程中判断左子树和右子树组成的子树是否出现过（用pair记录子树的情况，也就是左右子树，两个map，第一个map表示子树的pair与它的标号的映射，第二个map表示每种标号出现的次数）。

代码：

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <ctype.h>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
#define eps 1e-8
#define INF 0x7fffffff
#define maxn 100005
#define PI acos(-1.0)
#define seed 31//131,1313
typedef long long LL;
typedef unsigned long long ULL;
using namespace std;
int top=0;
map < pair < int , int > , int > M;
map < int , LL > N ;
int tree_a[maxn][2],tree_b[maxn][2],a[maxn],b[maxn];
LL ans;
void dfs_a(int u)
{
pair < int , int > p;
int left,right;
left=tree_a[u][0],right=tree_a[u][1];
if(left!=-1)
dfs_a(left);
if(right!=-1)
dfs_a(right);
if(left==-1)
p.first=-1;
else p.first=a[left];
if(right==-1)
p.second=-1;
else p.second=a[right];
if(M.find(p)==M.end())
{
M[p]=top;
a[u]=top;
N[top]=1;
top+=3;
}
else
{
a[u]=M[p];
N[M[p]]++;
}
}
void dfs_b(int u)
{
pair <int ,int > p ;
int left=tree_b[u][0],right=tree_b[u][1];
if(left!=-1)
dfs_b(left);
if(right!=-1)
dfs_b(right);
if(left==-1)
p.first=-1;
else p.first=b[left];
if(right==-1)
p.second=-1;
else p.second=b[right];
if(M.find(p)!=M.end())
{
ans+=N[M[p]];
b[u]=M[p];
}
}
void init()
{
top=0;
ans=0;
N.clear();
M.clear();
for(int i=0;i<=100000;i++)
a[i]=-2,b[i]=-2;
}
int main()
{
int T,n,m;
scanf("%d",&T);
while(T--)
{
init();
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++)
scanf("%d%d",&tree_a[i][0],&tree_a[i][1]);
for(int i=1; i<=m; i++)
scanf("%d%d",&tree_b[i][0],&tree_b[i][1]);
dfs_a(1);
dfs_b(1);
printf("%lld\n",ans);
}
return 0;
}