ZOJ - 3602 Count the Trees hash标记

题目链接：点击查看

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

题解：第一个树扫一遍标记一下，然后第二个树再扫一遍统计一下即可，用pair<左子树编号，右子树编号> 标记，之前用map二维标记的不知道为啥段错误。。。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int a[N][2],b[N][2];
map<pair<int,int>, int > mp;
int num[N];
int mpid,n,m;
ll ans;
int dfs1(int u)
{
	int f1,f2,id;
	if(a[u][0]!=-1)
		f1=dfs1(a[u][0]);
	else f1=-1;
	
	if(a[u][1]!=-1)
		f2=dfs1(a[u][1]);
	else f2=-1;
	if(mp[make_pair(f1,f2)]==0)mp[make_pair(f1,f2)]=++mpid;
	id=mp[make_pair(f1,f2)];
	num[id]++;
	return id;
}
int dfs2(int u)
{
	int f1,f2,id;
	if(b[u][0]!=-1)
		f1=dfs2(b[u][0]);
	else f1=-1;
	
	if(b[u][1]!=-1)
		f2=dfs2(b[u][1]);
	else f2=-1;
	if(mp[make_pair(f1,f2)]==0) mp[make_pair(f1,f2)]=0;
	id=mp[make_pair(f1,f2)];
	ans+=num[id];
	return id;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		ans=0;
		mpid=0;
		mp.clear();
		for(int i=1;i<=100000;i++)num[i]=0;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)scanf("%d%d",&a[i][0],&a[i][1]);
		for(int i=1;i<=m;i++)scanf("%d%d",&b[i][0],&b[i][1]);
		dfs1(1);
		dfs2(1);
		printf("%lld\n",ans);
	}
	return 0;
}