剑指Offer 树的子结构

最新推荐文章于 2022-05-24 17:36:34 发布

原创最新推荐文章于 2022-05-24 17:36:34 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 同时被 2 个专栏收录

80 篇文章

订阅专栏

剑指Offer

80 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树是否为另一棵二叉树的子结构。通过前序遍历，寻找匹配的根节点，并进一步验证子树的结构和值是否完全匹配。

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

思路:

1、原二叉树走前序遍历，试图发现哪个节点的值和被判断子树的根节点相同；

如果一直到最后也没有找到那么肯定不是

2、如果找到了，就两个二叉树一起前序遍历，试图发现两个二叉树同时遍历完成，且同样的左右子树遍历过程中的节点值均相同，同时遍历完成说明有相同的形状，左右子树遍历过程中的节点值均相同说明两个二叉树不仅形状一样，值也是一样，则可以认为是子树。

3. 如果找到与原树与根结点相同，但不是子树，此时原树往左右结点依次遍历。重复1、2步骤。

牛客网AC代码：

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
private:
bool flag;
public:
     bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1 == NULL || pRoot2 == NULL)
            return false;
         flag = false;
         HasSubtreeCore(pRoot1,pRoot2);
         return flag;
    }
    void HasSubtreeCore(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(flag == true)//剪枝
            return ;
        if(pRoot1 == NULL)
            return ;
        if(pRoot1->val == pRoot2->val)
        {
            if(IsSubtree(pRoot1,pRoot2)){
                flag = true;
                return;
            }
        }
        HasSubtreeCore(pRoot1->left,pRoot2);
        HasSubtreeCore(pRoot1->right,pRoot2);
    }
    
    bool IsSubtree(TreeNode* pRoot1, TreeNode* pRoot2){
        if(pRoot1==NULL && pRoot2!=NULL)
            return false;
        if(pRoot1==NULL && pRoot2==NULL)
            return true;
        if(pRoot1!=NULL && pRoot2==NULL)
            return true;
        if(pRoot1->val != pRoot2->val)
            return false;
        
        return IsSubtree(pRoot1->left,pRoot2->left)
            && IsSubtree(pRoot1->right,pRoot2->right);
    }
};