我在这道题上栽过的坑

树的子结构

http://www.nowcoder.com/questionTerminal/6e196c44c7004d15b1610b9afca8bd88

第一坑
首先我以为,只要逐一比较各个节点就好了。作为一棵被比较的子树,它应该完全和原树的某个子树相同(包含所有叶子节点),而不是组成某子树的一部分(不包含该子树的所有叶子节点)。所以我写了:

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

结果显然是错了,因为错的例子中有一个是
{8,8,7,9,2,#,#,#,#,4,7},{8,9,2}
对应输出应该为:
true
你的输出为:
false

Okay,我明白了,你允许root1的子树有多余的叶子节点。

第二坑
于是我改掉了上面代码的第4行,变成了

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

蛤,结果又错了!
这,究竟是人性的扭曲,还是道德的沦丧??
一看题目, 噢

(ps:我们约定空树不是任意一个树的子结构)

原来如此,失敬失敬!!
本来打算写一个漂亮的函数递归自身就行,现在看来不得不把递归的这个写成另一个函数了。否则我无法区分 root2==null 的情况。所以最终答案为。

public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root2==null) return false;
        return hasSubtree(root1,root2);
    }

    boolean hasSubtree(TreeNode root1, TreeNode root2){
        if(root1==null&&root2==null) return true;
        if(root1==null) return false;
        if(root2==null) return true;
        if(root1.val==root2.val&&
           hasSubtree(root1.left, root2.left)&&
           hasSubtree(root1.right, root2.right))
            return true;
        return hasSubtree(root1.left, root2)||hasSubtree(root1.right,root2);
    }
}
全部评论

相关推荐

10-11 15:42
皖西学院 Java
青鱼LINK:我硕士,也是java0面试,吾道不孤
点赞 评论 收藏
分享
评论
5
收藏
分享
牛客网
牛客企业服务