没有问题,但是不是最优的题解

树的子结构

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

两年前做过这道题目,这次发现不会做了,一直没有想到合适的解法(其实有一点思路,但是不会写),然后查看题解,也看了一下之前自己的题解,没有发现正确答案。其中一个反例:root2如果是root1的叶子节点(也就是一个只有根节点的二叉树),其中root1的深度大于等于3,那么其他答案的结果返回false,实际上应该是true。
我的解法:获取所有节点,依次判断是否是子结构。
不足:这块是有多余的判断,应该可以优化,正确的题解应该跟dfs或者dp有关,主要优化点在于保存已校验的状态

    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if (root1 == null || root2 == null)
            return false;
        //遍历root1
        Stack<TreeNode> stack = new Stack<>();
        getTreeNodes(root1, stack);
        while (!stack.isEmpty()) {
            boolean flag = hasSubtree(stack.pop(), root2);
            if (flag) {
                return flag;
            }
        }
        return false;
    }

    private void getTreeNodes(TreeNode root, Stack<TreeNode> stack) {
        if (root == null){
            return;
        }
        stack.push(root);
        getTreeNodes(root.left, stack);
        getTreeNodes(root.right, stack);
    }

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

相关推荐

用微笑面对困难:你出于礼貌叫了人一声大姐,大姐很欣慰,她真把你当老弟
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务