Java实现剑指Offer27: 树的子结构以及遇到的坑

题目

输入两颗二叉树A和B，判断B是不是A的子结构。例如，右边的树是左边的树的子结构。

分析

1.在大树中找到和小树根节点相同的节点。

2.然后以此节点为根节点，在大树上往下搜索对比小书左右节点是否相同，不同则返回false

3.如果2步中最后返回false 回到第一步，从大树的左子树找和小树根节点相同的节点。

4.如果3步中最后返回false 回到第一步，从大树的右子树找和小树根节点相同的节点。

代码思路

两个递归解决该问题。首先创建一个名为HasSubtree的方法来判断大树中有没有和小树根节点相同的节点，若有，执行DoesTree1HaveTree2方法判断在大树上往下搜索对比小书左右节点是否完全相同；若没有，继续遍历大树直到找到与小树根节点相等的节点。

有问题的解法

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

        
    public boolean DoesTree1HaveTree2(TreeNode root1,TreeNode root2){
    	if (root1 == null ) return false;
        if(root2 == null) return true;
        if(root1.val == root2.val) 
        return DoesTree1HaveTree2(root1.left,root2.left)&&DoesTree1HaveTree2(root1.right,root2.right);
        return false;

问题分析

错误1. 在DoesTree1HaveTree2方法中，一定要先判断小树是否为空，而不能先判断大树为空，因为可能出现刚好大树的尾部是小树，二者同时结束的情况。

错误2：if (root1.val == root2.val) return DoesTree1HaveTree2(root1,root2);
假如大树某一节点与小树根节点一致，而余下部分不一致，那么在还没有继续向下遍历大树的时候就已经返回了false，显然是不对的。

修改后的代码

1.在HasSubtree方法中加入flag，使其继续判断大树余下的部分。

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

2.短路特性

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