https://pintia.cn/problem-sets/15/problems/711
给定两棵树T1和T2。如果T1可以通过若干次左右孩子互换就变成T2,则我们称两棵树是“同构”的。例如图1给出的两棵树就是同构的,因为我们把其中一棵树的结点A、B、G的左右孩子互换后,就得到另外一棵树。而图2就不是同构的。
图1
图2
现给定两棵树,请你判断它们是否是同构的。
输入格式:
输入给出2棵二叉树树的信息。对于每棵树,首先在一行中给出一个非负整数N (≤),即该树的结点数(此时假设结点从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 <string.h> #define MAXLEN 10 typedef struct node { char letter; int father; }Node; int InputArr(Node tr[]) { int len; char c,l,r; scanf("%d",&len); getchar(); for(int i=0;i<len;i++) { scanf("%c %c %c",&tr[i].letter,&l,&r); getchar(); //printf("%c %c %c ",tr[i].letter,l,r); if(l!='-') { tr[l-'0'].father = i; } if(r!='-') { tr[r-'0'].father = i; } } return len; } bool isSame(Node tr1[], int len1, Node tr2[], int len2) { if(len1==1 && len2==1) return tr1[0].letter==tr2[0].letter; if(len1 != len2) return false; for(int i=0;i<len1;i++) { for(int j=0;j<len2;j++) { if(tr1[i].letter == tr2[j].letter) { if(tr1[i].father==-1 || tr2[i].father==-1) { } else if(tr1[tr1[i].father].letter != tr2[tr2[j].father].letter) { // printf("%c %c %c %c ", tr1[i].letter,tr2[j].letter,tr1[tr1[i].father].letter,tr2[tr2[j].father].letter); return false; } break; } } } return true; } void init(Node arr[],int len) { for(int i=0;i<len;i++) arr[i].father = -1; } int main() { Node tr1[MAXLEN],tr2[MAXLEN]; init(tr1,MAXLEN); init(tr2,MAXLEN); int len1 = InputArr(tr1); int len2 = InputArr(tr2); printf( isSame(tr1,len1,tr2,len2) ? "Yes" : "No"); // printf("=============== "); // for(int i=0;i<len1;i++) // { // printf("%c %d ",tr1[i].letter,tr1[i].father); // } // printf("=============== "); // for(int i=0;i<len2;i++) // { // printf("%c %d ",tr2[i].letter,tr2[i].father); // } return 0; }