给定两个二叉树s和t,判断t是否为s的子树。
方法一:进行对比时,s不是从根节点开始的,而t是从根节点开始的,因此,可以先计算t的高度,然后在s中找出所有高度和t相同的子树,然后对比它们是否相同即可。
class Solution {
int height = 0;
LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
public boolean isSubtree(TreeNode s, TreeNode t) {
height = height(t);
height(s);
while(!queue.isEmpty()){
TreeNode root = queue.poll();
if(isSame(root, t))
return true;
}
return false;
}
public int height(TreeNode root){
if(root == null)
return 0;
int h = Math.max(height(root.left), height(root.right)) + 1;
if(height == h) //计算t的高度时,height=0,肯定不会相等。计算s的高度时,将高度等于height的所有节点装进队列。
queue.offer(root);
return h;
}
public boolean isSame(TreeNode s, TreeNode t){
if(s == null || t == null)
return s == t;
return isSame(s.left, t.left) && isSame(s.right, t.right) && s.val == t.val;
}
}
方法二:直接对s的左子树和右子树进行递归判断,强制循环所有节点。
这种方法时间复杂度较高,176个测试案例,方法一的时间是16ms,方法二是26ms
public class Solution {
public boolean isSubtree(TreeNode s, TreeNode t) {
if (s == null) return false;
if (isSame(s, t)) return true;
return isSubtree(s.left, t) || isSubtree(s.right, t);
}
private boolean isSame(TreeNode s, TreeNode t) {
if (s == null && t == null) return true;
if (s == null || t == null) return false;
if (s.val != t.val) return false;
return isSame(s.left, t.left) && isSame(s.right, t.right);
}
}
方法三:把s和t转化成字符串,直接判断s是否包含t,虽然这种方法看起来是O(n),但实际上涉及到栈和字符串操作,测试案例的时间为39ms。
public boolean isSubtree(TreeNode s, TreeNode t) {
String spreorder = generatepreorderString(s);
String tpreorder = generatepreorderString(t);
return spreorder.contains(tpreorder) ;
}
public String generatepreorderString(TreeNode s){
StringBuilder sb = new StringBuilder();
Stack<TreeNode> stacktree = new Stack();
stacktree.push(s);
while(!stacktree.isEmpty()){
TreeNode popelem = stacktree.pop();
if(popelem==null)
sb.append(",#"); // Appending # inorder to handle same values but not subtree cases
else
sb.append(","+popelem.val);
if(popelem!=null){
stacktree.push(popelem.right);
stacktree.push(popelem.left);
}
}
return sb.toString();
}