给你两棵二叉树 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
解题思路:
本题使用递归+迭代的方法解决。
首先定一个函数,用来判断两棵树是否相同。然后我们在主函数中使用迭代法,这里使用的是前序遍历,来遍历整棵树,如果出现相同的树则返回。
/**
* 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) {
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pop();
if (isSame(node, subRoot)) {return true;}
if (node.right != null) {stack.push(node.right);}
if (node.left != null) {stack.push(node.left);}
}
return false;
}
public boolean isSame(TreeNode p, TreeNode q) {
if (p == null && q == null) {return true;}
else if (p == null && q != null) {return false;}
else if (p != null && q == null) {return false;}
else if (p.val != q.val) {return false;}
return isSame(p.left, q.left) && isSame(p.right, q.right);
}
}