PTA 7-34 樹的同構 (25分)
阿新 • • 發佈:2021-02-18
技術標籤:演算法
給定兩棵樹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<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;
}