利用短路特性,参考https://www.nowcoder.com/questionTerminal/6e196c44c7004d15b1610b9afca8bd88?f=discussion
题目:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
public class Solution {
public boolean HasSubtree1(TreeNode root1,TreeNode root2) {
if(root2==null) {
return true;
}
if(root1==null) {
return false;
}
if(root1.val==root2.val) {
return HasSubtree1(root1.left, root2.left)&&HasSubtree1(root1.right, root2.right);
}
return false;
}
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(root1==null||root2==null) {
return false;
}
return HasSubtree1(root1, root2)||HasSubtree(root1.left, root2)||HasSubtree(root1.right, root2);
}
}
分析一下:
如果B是A的子树肯定从A树的某个点开始,到A树的底部都是一样的,所以需要遍历所以的节点,找到相等的那个点,
这还有一个问题:相等的那个点有可能有多个,不知道B树是从哪个点开始的,所以必须要遍历所有的点。
HasSubtree1(root1, root2)||HasSubtree(root1.left, root2)||HasSubtree(root1.right, root2);
这段代码运用的是短路特性,就是每个节点都要和B的根节点比较,相同就去HasSubtree1
方法里对比B树的所有结构是否和A对应上。
if(root2==null) {
return true;
}
if(root1==null) {
return false;
}
上面这段代码是为了到底部判断用的,有可能B是A的内部而不是子结构,再通俗点就是B节点和A节点完全对应上了,但是A树还可以继续往下,而B树已经没有节点了。
if(root1.val==root2.val) {
return HasSubtree1(root1.left, root2.left)&&HasSubtree1(root1.right, root2.right);
}
上面这段代码用的是&&,因为是子结构,必须全部吻合才行。