代码随想录——打家劫舍Ⅱ(Leetcode 213)
题目链接
动态规划
动态规划思路:
定义一个数组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]);}
}