题目链接:点击查看
题意:给出两棵二叉树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;
}