题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
分析
一开始我把子结构理解成一棵树得在另一颗树的末端才算是子结构,后来看了讨论区发现可以位于树的中间位置。
深度遍历二叉树知道找到A的一个结点值等于B的根节点的值,然后判断从这两个结点开始对比,B是否在A中。
妈耶,居然有点难以确认自己写的判断两棵树是否相同的代码部分是不是正确了…不过这里的写法是针对这一题的,如果是直接对比两颗二叉树是否相等,开始的条件会有所变动。
时间复杂度
O(n)
代码
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
Stack<TreeNode> s = new Stack<TreeNode>();
if(root1 == null)return false;
if(root2 == null)return false;
//DFS,找到相同值得结点,再看从这个结点起,两树结构是否相同
s.push(root1);
while(!s.isEmpty()){
TreeNode p = s.pop();
if(p.val == root2.val){
if(ifSame(p,root2))return true;
}
if(p.left!=null)s.push(p.left);
if(p.right!=null)s.push(p.right);
}
return false;
}
boolean ifSame(TreeNode root1,TreeNode root2){
if(root2 != null && root1 == null)return false;
if(root2 == null)return true;
//短路特性,若两者不想等,不用递归了
return root1.val==root2.val &&
ifSame(root1.left,root2.left) &&
ifSame(root1.right,root2.right);
}
}