【刷题】牛客CD170-判断 t1 树是否包含 t2 树全部的拓扑结构

题目描述 给定彼此独立的两棵二叉树，判断 t1 树是否包含 t2 树全部的拓扑结构。 设 t1 树的边集为 E1，t2 树的边集为 E2，若 E2 是 E1 的子集，则表示 t1 树包含 t2 树全部的拓扑结构。 输入描述: 第一行输入两个整数 n 和 root，n 表示二叉树t1 的总节点个数，root 表示二叉树 t1 的根节点。

以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa
没有左儿子，rch同理)

第 n+2 行输入两个整数 m 和 root，n 表示二叉树 t2 的总节点个数，root 表示二叉树 t2 的根节点。

以下 m 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa
没有左儿子，rch同理) 输出描述: 如果 t1 树包含 t2 树全部的拓扑结构，则输出 “true”，否则输出 “false”。 示
例1
输入
10 1
1 2 3
2 4 5
4 8 9
8 0 0
9 0 0
5 10 0
10 0 0
3 6 7
6 0 0
7 0 0
4 2
2 4 5
5 0 0
4 8 0
8 0 0
输出
true

需要注意的是如下图的这种情况，t2已经无法再继续遍历，但是t1仍有节点，也是符合题目条件的。

#include<bits/stdc++.h>
using namespace std;
struct TreeNode{
  int value;
  TreeNode*left,*right;
  TreeNode(int value_):value(value_),left(NULL),right(NULL){}
};
map<int,TreeNode*>mp1;
map<int,TreeNode*>mp2;
bool isSubTree(TreeNode*root1,TreeNode*root2){
    if(root1==NULL||root2==NULL){
        if(root1==NULL&&root2==NULL)
            return 1;
        else if(root1!=NULL&&root2==NULL){
            return 1;
        }
        else return 0;
    }
    bool cur = (root1->value==root2->value)?1:0;
    bool left = isSubTree(root1->left, root2->left);
    bool right = isSubTree(root1->right, root2->right);
    return cur&&left&&right;
}
TreeNode* createRoot(int value,map<int,TreeNode*>&mp){
    TreeNode* cur = new TreeNode(value);
    if(mp.count(value)==0){
        cur = new TreeNode(value);
        mp[value] = cur;
    }else{
        cur = mp[value];
    }
    return cur;
}
int main(){
    int n,r1;
    cin>>n>>r1;
    TreeNode* root1 = new TreeNode(r1);
    mp1[r1] = root1;
    for(int i=0;i<n;i++){
        int fa,lch,rch;
        scanf("%d %d %d",&fa,&lch,&rch);
        TreeNode* cur = createRoot(fa,mp1);
        if(lch != 0){
            cur->left = createRoot(lch,mp1);
        }else{
            cur->left = NULL;
        }
        if(rch != 0){
            cur->right = createRoot(rch,mp1);
        }else{
            cur->right = NULL;
        }
    }
    int m,r2;
    cin>>m>>r2;
    TreeNode* root2 = new TreeNode(r2);
    mp2[r2] = root2;
    for(int i=0;i<m;i++){
        int fa,lch,rch;
        scanf("%d %d %d",&fa,&lch,&rch);
        TreeNode* cur = createRoot(fa,mp2);
        if(lch != 0){
            cur->left = createRoot(lch,mp2);
        }else{
            cur->left = NULL;
        }
        if(rch != 0){
            cur->right = createRoot(rch,mp2);
        }else{
            cur->right = NULL;
        }
    }
    if(isSubTree(mp1[r2], root2))
        cout<<"true"<<endl;
    else
        cout<<"false"<<endl;
    return 0;
}