题目链接

打家劫舍II

题目描述

image.png

实现代码

思路：这题与前面的打家劫舍的区别就在于：对于超过2个住户的情况，第一户和最后一户人家只能两者选择一个进行偷窃；
记dp[i]表示有i个住户的情况能窃取到的最大金额，可以看出如果选择第一户，那么dp[i]就是在0 - len-2中去选，如果选择最后一户，那么dp[i]就是在1 - len-1中去选，因此可得对应的状态转移方程：

result = Math.max(dynamicPlan(nums, 0, len-2), dynamicPlan(nums, 1, len-1));

dynamicPlan(int[] nums, int startIdx, int endIdx) { dp[i] = Math.max(dp[i-1], dp[i-2] + nums[startIdx + i]); }

实现代码如下：

  1. class Solution {
  2.     public int rob(int[] nums) {
  4.         int len = nums.length;
  5.         if(len == 1) {
  6.             return nums[0];
  7.         }
  8.         if(len == 2) {
  9.             return Math.max(nums[0], nums[1]);
  10.         }
  11.         return Math.max(dynamicPlan(nums, 0, len-2), dynamicPlan(nums, 1, len-1));
  12.     }
  14.     private int dynamicPlan(int[] nums, int startIdx, int endIdx) {
  16.         int[] dp = new int[endIdx - startIdx + 1];
  17.         dp[0] = nums[startIdx];
  18.         dp[1] = Math.max(nums[startIdx], nums[startIdx+1]);
  20.         for(int i=2; i<endIdx - startIdx + 1; i++) {
  21.             dp[i] = Math.max(dp[i-1], dp[i-2] + nums[i+startIdx]);
  22.         }
  24.         return dp[endIdx - startIdx];
  26.     }
  27. }