问题描述
LeetCode 572. 另一棵树的子树 (opens in a new tab),难度简单。
给你两棵二叉树 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
题解
递归
解题思路:
-
如果两个树都为空,则它们相同。
-
如果其中一个树为空,而另一个不为空,则它们不同。
-
比较两个树的根节点值。如果根节点值相同,递归检查左子树和右子树。
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);
}
}