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