PTA-树的同构

给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2，则我们称两棵树是“同构”的。例如图1给出的两棵树就是同构的，因为我们把其中一棵树的结点A、B、G的左右孩子互换后，就得到另外一棵树。而图2就不是同构的。

图1

图2

现给定两棵树，请你判断它们是否是同构的。

输入格式:

输入给出2棵二叉树树的信息。对于每棵树，首先在一行中给出一个非负整数$N$ ($\le 10$)，即该树的结点数（此时假设结点从0到$N-1$编号）；随后$N$行，第$i$行对应编号第$i$个结点，给出该结点中存储的1个英文大写字母、其左孩子结点的编号、右孩子结点的编号。如果孩子结点为空，则在相应位置上给出“-”。给出的数据间用一个空格分隔。注意：题目保证每个结点中存储的字母是不同的。

输出格式:

如果两棵树是同构的，输出“Yes”，否则输出“No”。

输入样例1（对应图1）：

8A 1 2B 3 4C 5 -D - -E 6 -G 7 -F - -H - -8G - 4B 7 6F - -A 5 1H - -C 0 -D - -E 2 -

输出样例1:

Yes

输入样例2（对应图2）：

8B 5 7F - -A 0 3C 6 -H - -D - -G 4 -E 1 -8D 6 -B 5 -E - -H - -C 0 2G - 3F - -A 1 4

输出样例2:

No

这道题之前试过把它还原成原来的模样，然后进行比较，但是后面发觉其实没有必要，因为这些节点是在已有，所以不需要建立新的树，直接存着节点的左右子树的下标就可以了。如果没有孩子，那么用下标为-41做标记。

#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>#include <string>#include <cmath>#include <algorithm>#include <map>using namespace std;struct  node{char data[2];int  L,R;}Edge1[20],Edge2[20];int vist[20],root1,root2;int judge(int root1,int root2){if(root1 == -1 && root2 == -1)return 1;if(root1 != -1 && root2 != -1)if(Edge1[root1].data[0] == Edge2[root2].data[0])if((judge(Edge1[root1].L,Edge2[root2].L)&&judge(Edge1[root1].R,Edge2[root2].R))||(judge(Edge1[root1].L,Edge2[root2].R)&&judge(Edge1[root1].R,Edge2[root2].L)))return 1;return 0;}int main(){int n,m;scanf("%d",&n);char strL[2],strR[2];memset(vist,0,sizeof(vist));for(int i = 0; i < 20; i++)Edge1[i].L = Edge1[i].R = Edge2[i].L = Edge2[i].R = -1;for(int i = 0; i < n; i++){scanf("%s %s %s",Edge1[i].data,strL,strR);if(strL[0] != '-'){Edge1[i].L = strL[0]-'0';vist[strL[0]-'0'] = 1;}elseEdge1[i].L = -1;if(strR[0] != '-'){Edge1[i].R = strR[0]-'0';vist[strR[0]-'0'] = 1;}elseEdge1[i].R = -1;}for(int i = 0; i < n; i++)if(!vist[i]){root1 = i;break;}scanf("%d",&m);memset(vist,0,sizeof(vist));for(int i = 0; i < m; i++){scanf("%s %s %s",Edge2[i].data,strL,strR);if(strL[0] != '-'){Edge2[i].L = strL[0]-'0';vist[strL[0]-'0'] = 1;}elseEdge2[i].L = -1;if(strR[0] != '-'){Edge2[i].R = strR[0]-'0';vist[strR[0]-'0'] = 1;}elseEdge2[i].R = -1;}for(int i = 0; i < n; i++)if(!vist[i]){root2 = i;break;}if(n != m)printf("No\n");else{if(judge(root1,root2))printf("Yes\n");elseprintf("No\n");}return 0;}