1. 程式人生 > 其它 >PTA 7-34 樹的同構 (25分)

PTA 7-34 樹的同構 (25分)

技術標籤:演算法

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

img

圖1

img

圖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<bits/stdc++.h>
using namespace std; const int N=1024+7; int n,l,r,sign,m; char ch,x,y;//字母、左孩子、右孩子 struct node{ char data='#'; char fa='#'; }T1[N],T2[N];//T1、T2兩棵數 bool cmp(node a,node b){ return a.data<b.data; } int main(){ cin>>n; for(int i=0;i<n;i++){ cin>>ch>>x>>y; T1[i].data=
ch;//儲存編號對應的該節點字母 if(x!='-'){ l=x-'0'; T1[l].fa=ch;//左孩子,儲存該編號對應的父親字母 } if(y!='-'){ r=y-'0'; T1[r].fa=ch;//右孩子,儲存該編號對應的父親字母 } } cin>>m; getchar(); for(int i=0;i<m;i++){ cin>>ch>>x>>y; T2[i].data=ch; if(x!='-'){ l=x-'0'; T2[l].fa=ch; } if(y!='-'){ r=y-'0'; T2[r].fa=ch; } } sort(T1,T1+n,cmp);//按自身編號對應的字母排序 sort(T2,T2+m,cmp); for(int i=0;i<n;i++){//判斷每個節點對應的本字母和父字母 if(T1[i].fa==T2[i].fa&&T1[i].data==T2[i].data)continue; else{ sign=1;//為無法轉換得到做標記 break; } } if(sign){ puts("No"); }else{ puts("Yes"); } return 0; }