问题描述
输入两棵二叉树A,B,判断B是不是A的子结构。(我们约定空树不是任意一个树的子结构)
假如给定A为{8,8,7,9,2,#,#,#,#,4,7},B为{8,9,2},2个树的结构如下,可以看出B是A的子结构
数据范围:
0 <= A的节点个数 <= 10000
0 <= B的节点个数 <= 10000
示例1
输入:
{8,8,7,9,2,#,#,#,#,4,7},{8,9,2}
复制
返回值:
true
复制
示例2
输入:
{1,2,3,4,5},{2,4}
复制
返回值:
true
复制
示例3
输入:
{1,2,3},{3,1}
复制
返回值:
false
思路
分别递归左右子树,判断是否符合条件。
关键代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
#include <cstddef>
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) {
if(pRoot2== NULL)return false;//空结点
if(pRoot1 == NULL && pRoot2 == NULL)return false;
if(pRoot1 == NULL && pRoot2 != NULL)return false;
if(pRoot1 != NULL && pRoot2 == NULL)return true;
//cout<<pRoot1->val<<"\t"<<pRoot2->val <<endl;
if(pRoot1->val == pRoot2->val){
//cout<<pRoot1->val<<"\t"<<pRoot2->val <<endl;
if(pRoot2->left == NULL && pRoot2->right == NULL)return true;
if(pRoot2->left != NULL && pRoot2->right == NULL)return HasSubtree(pRoot1->left, pRoot2->left);
if(pRoot2->left == NULL && pRoot2->right != NULL)return HasSubtree(pRoot1->right, pRoot2->right);
bool s1 = HasSubtree(pRoot1->left, pRoot2->left)&&
HasSubtree(pRoot1->right, pRoot2->right);
if(s1)return true;
}
//递归左右子树
if(pRoot1->left || pRoot1->right)
return HasSubtree( pRoot1->left,pRoot2)||HasSubtree(pRoot1->right ,pRoot2);
return false;
}
};