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

xiaoxiao2021-02-28  50

给定的结构是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; }}

转载请注明原文地址: https://www.6miu.com/read-2624509.html

最新回复(0)