题目描述 给定彼此独立的两棵二叉树,判断 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;
}