力扣572—另一个树的子树(思路＋代码)

最新推荐文章于 2023-12-06 23:35:23 发布

keild

最新推荐文章于 2023-12-06 23:35:23 发布

阅读量320

点赞数 2

分类专栏：力扣文章标签： leetcode 算法数据结构 java 职场和发展

本文链接：https://blog.youkuaiyun.com/m0_61761695/article/details/126496269

版权

力扣专栏收录该内容

20 篇文章

订阅专栏

题目要求：
在这里插入图片描述

思路：判断是否为子树，三种情况，①两树相同，②root的左子树和subroot相同，③root的右子树和subroot相同。
需要判断两个树的val值是否相同，重点在另一个树中找到和他子树一样的根节点(可以递归另一个树的左子树，或者右子树，每一次递归都会判断是否为子树)

1.创建一个方法，判断是否为相同树(相同树在前几篇博客中有写到)

public boolean isSameTree(TreeNode p, TreeNode q) {
            if(p != null && q == null || p == null && q != null){
                return false;
            }
            if(p == null && q == null){
                return true;
            }
            if(p.val != q.val){
                return false;
                //只能判断！= 如果== 那么可能左树相同但是右树不同，但是已经返回了true
            }
            //走到这里说明 p!=null&&q!=null&&q.val = p.val 
            return isSameTree(p.left,q.left) && isSameTree(p.right,q.right);
        }

2.如果root和subroot其中有一个不为空但是另一个为空，则不满足条件，如果为两个相同树，如果subroot和root左子树或者右子树相同，递归

public boolean isSubtree(TreeNode root, TreeNode subRoot) {
            if(root != null && subRoot == null || root == null && subRoot !=null){
                return false;
            }
            
            if(isSameTree(root,subRoot)){//每次递归都会判断当前的root和subroot是否为相同树
                return true;
            }
            //递归左树和右树 判断是否为相同树
            return isSubtree(root.left,subRoot) || isSubtree(root.right,subRoot);
    }

完整代码👇

class Solution {
        public boolean isSameTree(TreeNode p, TreeNode q) {
            if(p != null && q == null || p == null && q != null){
                return false;
            }
            if(p == null && q == null){
                return true;
            }
            if(p.val != q.val){
                return false;
            }
            return isSameTree(p.left,q.left) && isSameTree(p.right,q.right);
        }

       
    
        public boolean isSubtree(TreeNode root, TreeNode subRoot) {
            if(root != null && subRoot == null || root == null && subRoot !=null){
                return false;
            }
            
            if(isSameTree(root,subRoot)){
                return true;
            }
            return isSubtree(root.left,subRoot) || isSubtree(root.right,subRoot);
    }
}