剑指offer：树的子结构代码实现

最新推荐文章于 2023-12-13 00:15:00 发布

原创最新推荐文章于 2023-12-13 00:15:00 发布 · 474 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #树的子结构 #剑指offer

经典编程题专栏收录该内容

130 篇文章

订阅专栏

本文详细阐述了如何判断一棵二叉树是否为另一棵二叉树的子结构，通过递归方法深入理解算法逻辑。

题目描述

输入两颗二叉树A，B，判断B是不是A的子结构。

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)
    {
        bool result = false;

        if(pRoot1 != NULL && pRoot2 != NULL)
        {
            /* 如果这个节点相同，则判断是否是子树 */
            if(pRoot1->val == pRoot2->val)
            {
                result = hasTree1IncludeTree2(pRoot1,pRoot2);
            }

            /* 如果不是，接下来递归判断左子树 */
            if(result == false)
            {
                result = HasSubtree(pRoot1->left,pRoot2);
            }

            /* 如果不是，接下来递归判断右子树 */
            if(result == false)
            {
                result = HasSubtree(pRoot1->right,pRoot2);
            }
        }

        return result;
    }

    /* 判断pRoot2是否是pRoot1的子树 */
    bool hasTree1IncludeTree2(TreeNode* pRoot1, TreeNode* pRoot2)
    {

        /* 如果树2为空，则所有都匹配，查找成功 */
        if(pRoot2 == NULL)
        {
            return true;
        }

        if(pRoot1 == NULL)
        {
            return false;
        }

        if(pRoot1->val != pRoot2->val)
        {
            return false;
        }

        return hasTree1IncludeTree2(pRoot1->left,pRoot2->left) && 
                hasTree1IncludeTree2(pRoot1->right,pRoot2->right);
    }
};