假如不偷第一家,dp[1]=0
假如偷第一家,dp[1]=nums[0]
两种方案取最大值即可,注意两种方案的结果其实都包含了既不偷第一家和不偷第二家
代码如下:
public int rob (int[] nums) {
int len=nums.length;
int[]dp=new int[len+1];
dp[1]=nums[0];//偷第一家
for(int i=2;i<len;i++){
dp[i]=Math.max(dp[i-1],dp[i-2]+nums[i-1]);
}
int res=dp[len-1];
Arrays.fill(dp,0);
//不偷第一家
for(int i=2;i<=len;i++){
dp[i]=Math.max(dp[i-2]+nums[i-1],dp[i-1]);
}
}