7-2 树的同构 (25 分)

7-2 树的同构 (25 分)

给定两棵树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 <iostream>
#include <algorithm>
#include <cstdio>
#include <map>
#include <set>
#include <vector>
#include <cstring>
#include <queue>
#include <cmath>
#define FRER() freopen("in.txt","r",stdin);
#define FREW() freopen("out.txt","w",stdout);
#define mem(a,b) memset(a,b,sizeof(a));
#define go int T;scanf("%d",&T);for(int cas=1;cas<=T;cas++)
#define mod 1000000007
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
const int maxn = 100 + 7 ;
int n,m;
int FA,FB;
int vis[maxn];
struct Tree{
    int left,right;
    char data;
}A[maxn],B[maxn];
void Init(){
    for(int i=0;i<n;i++){
        A[i].left=A[i].right=B[i].left=B[i].right = n;
        A[i].data = B[i].data = '*';
    }
}
void BuildA(int& F,int len){
    memset(vis, 0, sizeof vis);
    char a,b,c;
    for(int i=0;i<len;i++){
        scanf(" %c %c %c",&a,&b,&c);
        A[i].data = a;
        if(b!='-'){
            A[i].left = b - '0';
            vis[A[i].left] = 1;
        }
        if(c!='-'){
            A[i].right = c - '0';
            vis[A[i].right] = 1;
        }
    }
    for(int i=0;i<len;i++){
        if(!vis[i]) F = i;
    }
}
void BuildB(int& F,int len){
    memset(vis, 0, sizeof vis);
    char a,b,c;
    for(int i=0;i<len;i++){
        scanf(" %c %c %c",&a,&b,&c);
        B[i].data = a;
        if(b!='-'){
            B[i].left = b - '0';
            vis[B[i].left] = 1;
        }
        if(c!='-'){
            B[i].right = c - '0';
            vis[B[i].right] = 1;
        }
    }
    for(int i=0;i<len;i++){
        if(!vis[i]) F = i;
    }
}
void printA(int i){
    if(i==n) return;
    cout<<A[i].data<<" ";
    printA(A[i].left);
    printA(A[i].right);
}
void printB(int i){
    if(i==m) return;
    cout<<B[i].data<<" ";
    printB(B[i].left);
    printB(B[i].right);
}
bool ok(int a,int b){
    if(n!=m) return  false;
    if(A[a].data!=B[b].data) return false;
    if(a==b&&a==n) return true;
    return ((ok(A[a].left, B[b].left)&&ok(A[a].right,B[b].right))||(ok(A[a].left,B[b].right)&&ok(A[a].right, B[b].left)));
    return true;
}
int main(){
    //ios::sync_with_stdio(false);
    //FRER();
    scanf("%d",&n);
    Init();
    BuildA(FA,n);
    scanf("%d",&m);
    BuildB(FB,m);
    printf("%s\n",ok(FA,FB)?"Yes":"No");
}

猜你喜欢

转载自blog.csdn.net/Insist_77/article/details/83794269