1. 程式人生 > >03-樹1 樹的同構-PTA

03-樹1 樹的同構-PTA

給定兩棵樹T1和T2。如果T1可以通過若干次左右孩子互換就變成T2,則我們稱兩棵樹是“同構”的。例如圖1給出的兩棵樹就是同構的,因為我們把其中一棵樹的結點A、B、G的左右孩子互換後,就得到另外一棵樹。而圖2就不是同構的。

 

 

圖1

圖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>

#define ElementType char
#define Tree int
#define MaxTree 10
#define Null -1

struct TreeNode{
	ElementType Element;
	Tree left;
	Tree right;
}T1[MaxTree],T2[MaxTree];

Tree BuildTree(struct TreeNode T[]);
int Isomorphic(Tree R1,Tree R2);

int main()
{
	Tree R1,R2;
	R1 = BuildTree(T1);  //建立二叉樹R1 
	R2 = BuildTree(T2);  //建立二叉樹R2
	/*比較R1 R2兩者是否同構*/
	if( Isomorphic(R1,R2) )  printf("Yes\n");
	else printf("No\n");
	return 0;
 } 
 
 Tree BuildTree(struct TreeNode T[])
 {
 	int i,N;
 	int Root = Null;
 	char cl,cr;
 	int check[MaxTree];
 	scanf("%d\n",&N);
 	if( N ){
 		for(i=0; i<N; i++)
 		    check[i] = 0;    //標記陣列清零       
 		for(i=0; i<N; i++){
 			scanf("%c %c %c",&T[i].Element,&cl,&cr);
 			getchar();  //接收掉末尾換行符 
 			if(cl != '-'){  /*左兒子存在*/
 				T[i].left = cl-'0';    //當前節點的左兒子的編號(指標) 
 				check[T[i].left] = 1;  //已被節點指標指向的結點肯定不是根節點,做下標記 
			 }else{
			 	T[i].left = Null; 
			 }
			if(cr != '-'){
 				T[i].right = cr-'0';    //當前節點的右兒子的編號(指標) 
 				check[T[i].right] = 1;  //已被節點指標指向的結點肯定不是根節點,做下標記 
			 }else{
			 	T[i].right = Null; 
			 } 
		 }
		for(i=0; i<N; i++){
			if( !check[i] )  break;  //找到根節點 (沒有任何節點指標指向的結點) 
		}
		Root = i;
	 }
	 return Root;
 }
 
int Isomorphic(Tree R1,Tree R2)
 {
 	if((R1 == Null) && (R2==Null))  //樹都為空樹,視為同構 
 	    return 1;
 	if(((R1 == Null)&&(R2 != Null)) || ((R1 != Null)&&(R2 == Null)))  //兩樹其中之一為空樹 
	    return 0;
	if(T1[R1].Element != T2[R2].Element)  //若兩樹都非空的話,比較根節點 
	    return 0;
	if((T1[R1].left == Null) && (T2[R2].left == Null))  //左子樹都為空,比較右子樹 
	    return Isomorphic(T1[R1].right,T2[R2].right);
    if((T1[R1].left != Null)&&(T2[R2].left != Null) && (T1[T1[R1].left].Element == T2[T2[R2].left].Element))
    /*左子樹非空,並且左子樹根節點相同,遞迴比較左右子樹*/ 
        return (Isomorphic(T1[R1].left,T2[R2].left) && Isomorphic(T1[R1].right,T2[R2].right));
    else
        return (Isomorphic(T1[R1].left,T2[R2].right) && Isomorphic(T1[R1].right,T2[R2].left));
 }