树的子结构

最新推荐文章于 2024-12-14 13:57:46 发布

wenlyq

最新推荐文章于 2024-12-14 13:57:46 发布

阅读量83

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer 二叉树

本文链接：https://blog.youkuaiyun.com/wenlyq/article/details/88089505

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

57 篇文章

订阅专栏

二叉树

18 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树是否是另一棵二叉树的子结构。通过递归方法，首先检查两树根节点是否相同，然后分别检查左子树和右子树是否匹配。

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

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

1、首先设置标志位result = false，一旦匹配成功result就设为true，剩下的代码不会执行，如果匹配不成功，默认返回false。
2、递归思想，如果根节点相同则递归调用DoesTree1HaveTree2（），如果根节点不相同，则判断Tree1的左子树和Tree2是否相同，再判断右子树和Tree2是否相同。
3、注意null的条件，HasSubTree中，如果两棵树都不为空才进行判断。
DoesTree1HaveTree2中，
Tree2为空，则说明第二棵树遍历完了，即匹配成功；
Tree1为空有两种情况：（1）如果tree1为空&&tree2不为空，说明不匹配；（2）如果tree1为空，tree2为空，说明匹配。因此只要tree2为空，则匹配

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        res=False
        root1=pRoot1
        root2=pRoot2
        if(root1!=None and root2!=None)://root1、root2不为空才判断
            if root1.val==root2.val://根值相同，判断是否为子树
                res=self.DoesTree1HaveTree2(root1,root2)
            if (not res)://判断是否为左子树的子结构
                res=self.DoesTree1HaveTree2(root1.left,root2)
            if (not res)://判断是否为右子树的子结构
                res=self.DoesTree1HaveTree2(root1.right,root2)
            return res
    def DoesTree1HaveTree2(self,root1,root2):
        if root1==None and root2!=None:
            return False
        if root2==None:
            return True
        if root1.val!=root2.val:
            return False
        return self.DoesTree1HaveTree2(root1.left,root2.left) and self.DoesTree1HaveTree2(root1.right,root2.right)

   TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        boolean res=false;
        if(root1!=null && root2!=null){
            if(root1.val==root2.val){
                res=DoesTree1HaveTree2(root1,root2);
            }
            if(!res) res=DoesTree1HaveTree2(root1.left,root2);
            if(!res) res=DoesTree1HaveTree2(root1.right,root2);
        }
        return res;
    }
    public boolean DoesTree1HaveTree2(TreeNode root1,TreeNode root2){
        if (root1==null && root2!=null) return false;
        if (root2==null) return true;
        if(root1.val!=root2.val) return false;
        return DoesTree1HaveTree2(root1.left,root2.left) && DoesTree1HaveTree2(root1.right,root2.right);
    }
}