zoj 3602

链接

[https://vjudge.net/contest/293343#problem/C]

题意

给你两棵树。为有多少对子树是同构的

分析

就是简单的哈希吧。对于不同结构的树对应不同的哈希值
先明确同构是指它左右儿子都一样
所以这么映射 map<pair<int,int>,int> mp;最后一个值表示哈希值
然后先扫描第一棵树。统计每种结构的数量
扫描第二颗树的时候，就可以统计了。对于每种结构
就是两边的该结构数量的乘积，但这里不必统计第二颗树的了。你遇到该结构
就加就完事了。比如5*4就等于4个5相加

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+10;
map<pair<int,int>,int > mp;
//代表lson,rson对应的哈希值 
ll sum;
int Type;
int a[2][N],b[2][N];
ll num[N];
int dfs1(int x){
	int ls,rs;
	if(a[0][x]!=-1)
	ls=dfs1(a[0][x]);
	else ls=-1;
	if(a[1][x]!=-1)
	rs=dfs1(a[1][x]);
	else rs=-1;
	if(mp[make_pair(ls,rs)]==0) mp[make_pair(ls,rs)]=++Type;
	int curtype=mp[make_pair(ls,rs)];
	num[curtype]++;
	return curtype; 
}
int dfs2(int y){
	int ls,rs;
	if(b[0][y]!=-1)
	ls=dfs2(b[0][y]);
	else ls=-1;
	if(b[1][y]!=-1)
	rs=dfs2(b[1][y]);
	else rs=-1;
    int curtype=mp[make_pair(ls,rs)]; 
    sum+=num[curtype];
    return curtype;
}
int main(){
	int t,n,m;
	scanf("%d",&t);
	while(t--){
		memset(num,0,sizeof(num));
		mp.clear();
		sum=0,Type=0;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
		scanf("%d%d",&a[0][i],&a[1][i]);
		for(int j=1;j<=m;j++)
		scanf("%d%d",&b[0][j],&b[1][j]);
		dfs1(1);dfs2(1);
		printf("%lld\n",sum);
	}
	return 0;
}