算法_考考你——判断二叉树的子结构（剑指Offer真题）

原创于 2020-05-11 09:40:37 发布 · 187 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #二叉树

数据结构与算法同时被 2 个专栏收录

30 篇文章

订阅专栏

算法

25 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树B是否为另一棵二叉树A的子结构。通过递归方法，首先检查B是否为空，若为空则返回true；接着检查A是否为空，若为空则返回false；然后比较A与B的根节点值，如果相等，则继续递归检查A的左右子树是否包含B的左右子树。

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

判断B是否是A的子结构
递归模型描述

递归出口：

B为空，返回true
A为空，返回false
如果A 与 B相同
–>> 如果B的左子树是A的左子树的子结构
–>> 如果B的右子树是A的右子树的子结构，返回true

递归体：

判断B是否是A的左子树的子结构
判断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)
    {
        if (pRoot1 == nullptr || pRoot2 == nullptr)
            return false;
        return hasSubCore(pRoot1, pRoot2);
    }
    
    bool hasSubCore(TreeNode *pRoot1, TreeNode *pRoot2)
    {
        //递归出口
        if (pRoot2 == nullptr)
            return true;
        if (pRoot1 == nullptr)
            return false;
        if (pRoot1->val == pRoot2->val)
        {
            bool flag1 = hasSubCore(pRoot1->left, pRoot2->left);
            bool flag2 = hasSubCore(pRoot1->right, pRoot2->right);
             if (flag1 && flag2)
                return true;
        }
        bool has = hasSubCore(pRoot1->left, pRoot2);
        if (has)
            return true;
        return hasSubCore(pRoot1->right, pRoot2);
    }
};