判断树是否为二叉树的子树

给定的结构是HasSubtree(root1,root2)

即判断root1做根的树中是否含有root2节点做根的树。

二叉树的问题基本可以确定是用递归了,肯定是先递归遍历前者找到与右边树的根相等的节点,然后再判断对应子树。

即遍历前树,找到一个节点和root2相等,那么就把其作为根的子树列为考虑对象,接下来用另一个方法判断是不是。是就直接结束了,不是还要继续向下找下一个考虑对象而这里很显然用递归就可以了,这个节点不是,就去找其左右子树即可。

public boolean HasSubtree(TreeNode root1,TreeNode root2){
boolean result = false;
if(root1!=null && root2!=null){
if(root1.val == root2.val){
if(IsSubTree(root1,root2)){
result = true;
}
else{
result = HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
}
}
else{
result = HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
}
}
return result;
}

接下来是如果当前根与root2相等时要做的是不是子树的判断,即找到考虑对象后如何确定是不是要的子树:

也是相同的思路,先设个flag,后面就是给true的算法。

public boolean IsSubTree(TreeNode root1,TreeNode root2){
boolean result = false;
if(root1 != null){
if(root1.val == root2.val){
if(root2.left == null){
result = true;
}
else{
result = IsSubTree(root1.left,root2.left);
}
if(root2.right == null){
result = true && result;
}
else{
result = IsSubTree(root1.right,root2.right) && result;
}
}
}
return result;
}

按照写代码的思路来说把,我们先考虑的是

IsSubTree(root1.left,root2.left);

IsSubTree(root1.right,root2.right) 

这两句话,因为这两句话的意思就是检查两者左子树和右子树相等与否,如果都相等就是true的情况。

然后设置出口,出口肯定是遍历没的情况,但一定是root2遍历没才行。于是有了两个if else.这里别忘了判断右边的

时候并上left的结果,因为必须是两边都ok才能返回true的。

这样向下遍历就完成了,最外层的root1!=null是为了防止空指针异常,如果不加,为空后还要继续.left就报错了。

完整代码:

package exercise1;
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2){
boolean result = false;
if(root1!=null && root2!=null){
if(root1.val == root2.val){
if(IsSubTree(root1,root2)){
result = true;
}
else{
result = HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
}
}
else{
result = HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
}
}
return result;
}
public boolean IsSubTree(TreeNode root1,TreeNode root2){
boolean result = false;
if(root1 != null){
if(root1.val == root2.val){
if(root2.left == null){
result = true;
}
else{
result = IsSubTree(root1.left,root2.left);
}
if(root2.right == null){
result = true && result;
}
else{
result = IsSubTree(root1.right,root2.right) && result;
}
}
}
return result;
}
}

点赞