[LeetCode]198. House Robber I&II

最新推荐文章于 2025-04-14 23:34:52 发布

gqk289

最新推荐文章于 2025-04-14 23:34:52 发布

阅读量258

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/gqk289/article/details/54429418

LeetCode 专栏收录该内容

324 篇文章

订阅专栏

本文提供LeetCode上“打家劫舍”问题的两种情况的解决方案：一种是线性的房屋排列，另一种则是环状排列。对于线性排列的情况，通过动态规划的方法，在O(n)的时间复杂度和O(1)的空间复杂度下实现了最优解；对于环状排列，则采取分段求解的方式，避免同时抢劫首尾相邻的房屋。

https://leetcode.com/problems/house-robber/

显然空间复杂度O(1)的DP问题。rob和notRob记录当前位置抢或者不抢所能获得的最大收益。当前未抢 = max(上次抢了，上次未抢)；当前抢了 = 上次未抢 + 当前价值

public class Solution {
    public int rob(int[] nums) {
        if (nums == null) {
            return 0;
        }
        int rob = 0;
        int notRob = 0;
        for (int n : nums) {
            int curNotRob = notRob;
            notRob = Math.max(notRob, rob);
            rob = curNotRob + n;
        }
        return Math.max(rob, notRob);
    }
}

https://leetcode.com/problems/house-robber-ii/

如果数组成环。

那么首尾两个只能最多取一个，因此结果为1 ~ len - 1 和 0 ~ len - 2里面较大的那个

public class Solution {
    public int rob(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        return Math.max(rob(nums, 0, nums.length - 2), rob(nums, 1, nums.length - 1));
    }
    private int rob(int[] nums, int beg, int end) {
        if (nums.length == 1) {
            return nums[0];
        }
        int notRob = 0;
        int rob = 0;
        for (int i = beg; i <= end; i++) {
            int curNotRob = Math.max(notRob, rob);
            rob = notRob + nums[i];
            notRob = curNotRob;
        }
        return Math.max(rob, notRob);
    }
}