给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2,则我们称两棵树是“同构”的。例如图1给出的两棵树就是同构的,因为我们把其中一棵树的结点A、B、G的左右孩子互换后,就得到另外一棵树。而图2就不是同构的。
图1
图2
现给定两棵树,请你判断它们是否是同构的。
Input
输入数据包含多组,每组数据给出2棵二叉树的信息。对于每棵树,首先在一行中给出一个非负整数N (≤10),即该树的结点数(此时假设结点从0到N−1编号);随后N行,第i行对应编号第i个结点,给出该结点中存储的1个英文大写字母、其左孩子结点的编号、右孩子结点的编号。如果孩子结点为空,则在相应位置上给出”-”。给出的数据间用一个空格分隔。
注意:题目保证每个结点中存储的字母是不同的。
注意:题目保证每个结点中存储的字母是不同的。
Output
如果两棵树是同构的,输出“Yes”,否则输出“No”。
Example Input
8 A 1 2 B 3 4 C 5 - D - - E 6 - G 7 - F - - H - - 8 G - 4 B 7 6 F - - A 5 1 H - - C 0 - D - - E 2 -
Example Output
Yes
用结构体数组存储树的结点,一开始我把没有孩子的结点的孩子标号设为-1,在后面判断的时候简直麻烦死(=_=) 后来看了看以前AC的代码,原来是设为题目范围之外的一个数字(11),这样就不用额外判断了。正确的判断是看结点的左右孩子是否对应相等或者交叉相等,把输入和判断写成单个函数要比写在main函数里简便
#include <iostream>
#include <algorithm>
#include <cstdio>
const int N = 20;
struct node
{
char c;
int lchild, rchild;
}Tree1[N], Tree2[N];
bool judge(int i, int j)
{
if(Tree1[Tree1[i].lchild].c == Tree2[Tree2[j].lchild].c && Tree1[Tree1[i].rchild].c == Tree2[Tree2[j].rchild].c)
return true;
if(Tree1[Tree1[i].lchild].c == Tree2[Tree2[j].rchild].c && Tree1[Tree1[i].rchild].c == Tree2[Tree2[j].lchild].c)
return true;
return false;
}
bool check(int n)
{
int i, j;
for(i = 0; i < n; i++)
{
for(j = 0; j < n; j++)
{
if(Tree1[i].c == Tree2[j].c)
{
if(judge(i, j))
break;
else
return false;
}
}
if(j == n)
return false;
}
return true;
}
void Create(struct node Tree[], int n)
{
char s[2];
for(int i = 0; i < n; i++)
{
scanf("%s", s);
Tree[i].c = s[0];
scanf("%s", s);
if(s[0] != '-')
Tree[i].lchild = s[0] - '0';
else
Tree[i].lchild = 11;
scanf("%s", s);
if(s[0] != '-')
Tree[i].rchild = s[0] - '0';
else
Tree[i].rchild = 11;
}
}
int main()
{
int n, m;
while(~scanf("%d", &n))
{
Create(Tree1, n);
scanf("%d", &m);
Create(Tree2, m);
if(n != m)
printf("No\n");
else if(check(n))
printf("Yes\n");
else
printf("No\n");
}
return 0;
}