LeetCode-572. 另一棵树的子树

题目来源

572. 另一棵树的子树

题目详情

给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在，返回 true ；否则，返回 false 。

二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。

示例 1：

输入： root = [3,4,5,1,2], subRoot = [4,1,2]
输出： true

示例 2：

输入： root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2]
输出： false

提示：

• root 树上的节点数量范围是 [1, 2000]
• subRoot 树上的节点数量范围是 [1, 1000]
• -104 <= root.val <= 104
• -104 <= subRoot.val <= 104

题解分析

解法一：前序遍历

1. 我们可以把这道题分成两个部分，我们首先写一个方法专门用于判断两棵树是否相等。这个方法也是需要使用递归实现的，只有所有节点值都相同，这两棵树才是完全相等的。
2. 接着，我们可以依次判断根节点出发的树是否包含子树，如果不包含，我们再递归判断左右子树是否包含给定的子树。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public boolean isSubtree(TreeNode root, TreeNode subRoot) {
        if(root == null && subRoot == null){
            return true;
        }
        if(root == null && subRoot != null){
            return false;
        }

        if(judgeSubtree(root, subRoot)){
            return true;
        }

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