代码随想录——打家劫舍Ⅱ（Leetcode 213）

肉松拿铁

于 2024-07-31 11:15:45 发布

阅读量212

点赞数 5

分类专栏：算法题java版文章标签： leetcode 算法职场和发展数据结构 java

本文链接：https://blog.csdn.net/qq_46574748/article/details/140817747

版权

算法题java版专栏收录该内容

127 篇文章 0 订阅

订阅专栏

题目链接
在这里插入图片描述

动态规划

动态规划思路：

定义一个数组dp，其中dp[i]表示从第一个房间开始到第i个房间（不包括i+1）能偷到的最大金额
我们需要考虑两种情况：

不偷最后一个房间
不偷第一个房间
因为如果偷了第一个房间，就不能偷最后一个房间

class Solution {
    public int rob(int[] nums) {
        if (nums.length == 0) {
            return 0;
        }
        if (nums.length == 1) {
            return nums[0];
        }

        // 不偷最后一个房间的情况
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);
        for (int i = 2; i < nums.length - 1; i++) {
            dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
        }

        // 不偷第一个房间的情况
        int[] dp1 = new int[nums.length];
        dp1[1] = nums[1];
        for (int i = 2; i < nums.length; i++) {
            dp1[i] = Math.max(dp1[i - 2] + nums[i], dp1[i - 1]);
        }

        // 返回两种情况的最大值
        return Math.max(dp[nums.length - 2], dp1[nums.length - 1]);
    }
}