【LeetCode】337. 打家劫舍 III

最新推荐文章于 2024-09-20 23:05:22 发布

Mason97

最新推荐文章于 2024-09-20 23:05:22 发布

阅读量132

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/Mason97/article/details/106404067

版权

算法专栏收录该内容

78 篇文章 0 订阅

订阅专栏

在这里插入图片描述

思路是：递归解题，rob(根节点)可以分成两种情况：一定有根节点，一定没有根节点。其中，一定有根节点的情况，说明一定没有根节点的两个儿子。

这个方法是自上而下的，因而有许多重复的子问题。时间复杂度较高。

在这里插入图片描述

public class Solution {
    public int rob(TreeNode root) {
        // 用递归的思路
        if (root == null) return 0;
        int money = root.val;
        if (root.left != null) {
            money += rob(root.left.left) + rob(root.left.right);
        }
        if (root.right != null) {
            money += rob(root.right.left) + rob(root.right.right);
        }
        return Math.max(money, rob(root.right) + rob(root.left));
    }
}

用从下至上的方法解这道动态规划的题目。

可以看见时间的提升是926倍。

在这里插入图片描述

public class Solution {
    public int rob(TreeNode root) {
        // 应该由下至上，动态规划解题
        return method(root).val;
    }

    public TreeNode method(TreeNode node) {
        if (node == null){
            TreeNode tmp = new TreeNode(0);
            tmp.left = new TreeNode(0);
            tmp.right = new TreeNode(0);
            return tmp;
        }
        // node不为空
        node.left = method(node.left);
        node.right = method(node.right);
        // 分成两种情况：1，node被偷。2，node不被偷。
        node.val = Math.max(node.val +
                        node.left.left.val + node.left.right.val + node.right.left.val + node.right.right.val,
                node.left.val + node.right.val);
        return node;
    }
}