剑指Offer编程练习017——树的子结构

最新推荐文章于 2021-03-26 15:00:29 发布

zh921

最新推荐文章于 2021-03-26 15:00:29 发布

阅读量124

点赞数

分类专栏： C++ 算法

本文链接：https://blog.youkuaiyun.com/zhao_h/article/details/79725855

版权

C++ 同时被 2 个专栏收录

46 篇文章

订阅专栏

算法

29 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树是否为另一棵二叉树的子结构。通过递归方法，首先寻找两树根节点值相同的位置，然后比较其子树结构是否一致。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：
输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
}
};

思路：先找到树A中与树B根节点rootB相同的节点rootA，然后比较rootA和rootB两棵树，完全相等就说明树B是树A的子结构。

class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1==nullptr || pRoot2==nullptr)
            return false;
        if(pRoot1->val==pRoot2->val)//先找到根相等的
        {
            if(IsSub(pRoot1,pRoot2))//然后判断这个根下面的子树是否相等
                return true;
        }
        //先去左子树找，再去右子树找
        return HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2);
    }
    
    bool IsSub(TreeNode* p1,TreeNode* p2)
    {
        if(p2==nullptr)
            return true;
        if(p1==nullptr)
            return false;
        if(p1->val!=p2->val)
            return false;
        //左右子树必须严格匹配
        return IsSub(p1->left,p2->left) && IsSub(p1->right,p2->right);
    }
};