相同题型:100.相同的树 101.对称二叉树
给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。
二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。
提示:
- root 树上的节点数量范围是 [1, 2000]
- subRoot 树上的节点数量范围是 [1, 1000]
- -104 <= root.val <= 104
- -104 <= subRoot.val <= 104
解题思路:和他的相似类型题型是一个思路,同样拥有递归和迭代两种思路,不过迭代稍微麻烦一点,是在深度优先遍历的基础上进行是否为子树的判断。
递归,代码和提交结果如下:
//递归解法,深度优先遍历基础上,检查两颗树是否完全相同
class Solution {
public boolean isSubtree(TreeNode root, TreeNode subRoot) {
return dfs(root,subRoot);
}
private boolean dfs(TreeNode p, TreeNode q){
if(p == null){
return false;
}
return check(p,q) || dfs(p.left,q) || dfs(p.right,q);
}
private boolean check(TreeNode p, TreeNode q) {
if(p == null && q == null){
return true;
}else if(p != null && q == null){
return false;
}else if(q != null && p == null){
return false;
}else if(p.val != q.val){
return false;
}else{
return check(p.left,q.left) && check(p.right,q.right);
}
}
}
迭代,代码和提交结果如下:
class Solution {
public boolean isSubtree(TreeNode root, TreeNode subRoot) {
if(root == null){
return false;
}
if(subRoot == null){
return true;
}
Stack<TreeNode> stack = new Stack();
stack.push(root);
while(!stack.isEmpty()){
TreeNode temp = stack.pop();
if(subRoot.val == temp.val){
if(check(temp,subRoot)){
return true;
}
}
if(temp.right != null){
stack.push(temp.right);
}
if(temp.left != null){
stack.push(temp.left);
}
}
return false;
}
private boolean check(TreeNode temp, TreeNode subRoot){
Queue<TreeNode> queue = new LinkedList();
queue.offer(temp);
queue.offer(subRoot);
while(!queue.isEmpty()){
TreeNode temp1 = queue.poll();
TreeNode temp2 = queue.poll();
if(temp1 == null && temp2 == null){
continue;
}
if((temp1 != null && temp2 == null) || (temp1 == null && temp2 != null)){
return false;
}
if(temp1.val != temp2.val){
return false;
}
queue.offer(temp1.left);
queue.offer(temp2.left);
queue.offer(temp1.right);
queue.offer(temp2.right);
}
return true;
}
}
总结:虽然迭代写法很差,但是不失为一种写法,这个题感觉难度有中等难度,加油