题目
给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2,则我们称两棵树是“同构”的。例如图1给出的两棵树就是同构的,因为我们把其中一棵树的结点A、B、G的左右孩子互换后,就得到另外一棵树。而图2就不是同构的。
现给定两棵树,请你判断它们是否是同构的。
输入格式:
输入给出2棵二叉树树的信息。对于每棵树,首先在一行中给出一个非负整数N
(≤10),即该树的结点数(此时假设结点从0到N−1编号);随后N行,第i行对应编号第i个结点,给出该结点中存储的1个英文大写字母、其左孩子结点的编号、右孩子结点的编号。如果孩子结点为空,则在相应位置上给出“-”。给出的数据间用一个空格分隔。注意:题目保证每个结点中存储的字母是不同的。
输出格式:
如果两棵树是同构的,输出“Yes”,否则输出“No”。
输入样例1(对应图1):
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 -
输出样例1:
Yes
输入样例2(对应图2):
8
B 5 7
F - -
A 0 3
C 6 -
H - -
D - -
G 4 -
E 1 -
8
D 6 -
B 5 -
E - -
H - -
C 0 2
G - 3
F - -
A 1 4
输出样例2:
No
代码如下
#include <stdio.h>
#include <stdlib.h>
#define MaxTree 10
struct TreeNode
{
char data;
int lt;
int rt;
} T1[MaxTree], T2[MaxTree];
int BuildTree(struct TreeNode* T)
{
int i, N;
char cl, cr;
int check[100];
scanf("%d", &N); //输入需要建立树的长度
getchar();//一定要注意读取缓冲区的内容,否则下次读取字符时,会读取回车,导致不可控的错误,整棵树其实已经改变
//如果不确定前面具体的回车数量,可以使用 while(getchar()!='\n');可以一次性全部读取完毕
if (N)
{
for (i = 0; i < N; i++)
{
check[i] = 1; //初始化
}
for (i = 0; i < N; i++)
{
scanf("%c %c %c", &T[i].data, &cl, &cr);
getchar(); //注意
if (cl != '-')
{
T[i].lt = (int)(cl - '0');
check[T[i].lt] = -1;//因为编号为 (int)(cl - '0') 的结点为某结点的左子树,所以一定不是根结点
}
else
{
T[i].lt = -1;
}
if (cr != '-')
{
T[i].rt = (int)(cr - '0');
check[T[i].rt] = -1; //因为编号为 (int)(cr - '0') 的结点为某结点的右子树,所以一定不是根结点
}
else
{
T[i].rt = -1;
}
}
//判断是不是根节点,根节点没有双亲结点
for (i = 0; i < N; i++)
{
if (check[i] == 1)
{
return i; //返回根结点的编号
}
}
return -1;
}
return -1;
}
int Isomorphic(int R1, int R2)
{
if ((R1 == -1) && (R2 == -1))
{
return 1; // 两棵树都为空,同构
}
if (((R1 == -1) && (R2 != -1)) || ((R1 != -1) && (R2 == -1)))
{
return 0;//其中一个为空,不同构
}
if (T1[R1].data != T2[R2].data)
{
return 0; //根数据不同,不同构
}
if ((T1[R1].lt == -1) && (T2[R2].lt == -1))
{
return Isomorphic(T1[R1].rt, T2[R2].rt); //左子树为空,则判断右子树
}
if (((T1[R1].lt != -1) && (T2[R2].lt != -1)) && ((T1[T1[R1].lt].data) == (T2[T2[R2].lt].data)))
{
//两树左子树皆不为空,且值相等
//判断其子树
return (Isomorphic(T1[R1].lt, T2[R2].lt) && Isomorphic(T1[R1].rt, T2[R2].rt));
}
else//两树左子树有一个空 或者 皆不为空但值不相等
{
return (Isomorphic(T1[R1].lt, T2[R2].rt) && Isomorphic(T1[R1].rt, T2[R2].lt));
}
}
int main()
{
int R1, R2;
R1 = BuildTree(T1);
R2 = BuildTree(T2);
if (Isomorphic(R1, R2))
{
printf("Yes\n");
}
else
{
printf("No\n");
}
return 0;
}