leetcode
572. 另一棵树的子树

问题描述

LeetCode 572. 另一棵树的子树 (opens in a new tab),难度简单

给你两棵二叉树 rootsubRoot 。检验 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

题解

递归

解题思路:

  • 如果两个树都为空,则它们相同。

  • 如果其中一个树为空,而另一个不为空,则它们不同。

  • 比较两个树的根节点值。如果根节点值相同,递归检查左子树和右子树。

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