给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2,则我们称两棵树是“同构”的。
现给定两棵树,请你判断它们是否是同构的。
输入格式:
输入给出2棵二叉树树的信息。对于每棵树,首先在一行中给出一个非负整数N(≤10),即该树的结点数(此时假设结点从0到N−1编号);随后N行,第i行对应编号第i个结点,给出该结点中存储的1个英文大写字母、其左孩子结点的编号、右孩子结点的编号。如果孩子结点为空,则在相应位置上给出“-”。给出的数据间用一个空格分隔。注意:题目保证每个结点中存储的字母是不同的。
输出格式:
如果两棵树是同构的,输出“Yes”,否则输出“No”。
输入样例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:
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<cstdio>
#include<cstring>
using namespace std;
const int maxn=15;
int N,visit[maxn];
typedef struct Node Node;
typedef struct TreeNode TreeNode;
struct TreeNode{
char ch;
struct TreeNode *left,*right;
};
struct Node{
char ch;
char l,r;
};
Node node[maxn];
int FindRoot(){
int root=-1;
for(int i=0;i<N;i++){
scanf(" %c %c %c",&node[i].ch,&node[i].l,&node[i].r);
if(node[i].l!='-')
visit[node[i].l-'0']=1;
if(node[i].r!='-')
visit[node[i].r-'0']=1;
}
for(int i=0;i<N;i++){
if(visit[i]==0){
root=i;
break;
}
}
return root;
}
TreeNode* BuildTree(int num){
if(num<0)
return NULL;//空树直接返回NULL
TreeNode *root=new TreeNode;
root->ch=node[num].ch;
if(node[num].l=='-'){
root->left=NULL;
}
else{
root->left=BuildTree(node[num].l-'0');
}
if(node[num].r=='-')
root->right=NULL;
else{
root->right=BuildTree(node[num].r-'0');
}
return root;//空树返回-1
}
/*void post_order(TreeNode *R){
if(R==NULL)
return;
printf("%c\n",R->ch);
post_order(R->left);
post_order(R->right);
}*/
bool judge(TreeNode *root1,TreeNode *root2){
if(root1==NULL&&root2==NULL)
return true;
else if(root1==NULL||root2==NULL)
return false;
if(root1->ch!=root2->ch)
return false;
return (judge(root1->left,root2->left)&&judge(root1->right,root2->right))||(judge(root1->left,root2->right)&&judge(root1->right,root2->left));
}
int main(){
//freopen("in.txt","r",stdin);
TreeNode *root1,*root2;
memset(visit,0,sizeof(visit));
scanf("%d",&N);
int Root=FindRoot();
root1=BuildTree(Root);
//post_order(root1);
memset(visit,0,sizeof(visit));
scanf("%d",&N);
Root=FindRoot();
root2=BuildTree(Root);
//post_order(root2);
if(judge(root1,root2)){
puts("Yes");
}
else{
puts("No");
}
return 0;
}