假设有n个房子,走到n+1位置时的前一个状态可能是第n个房子或第n-1个房子,但中间可能会有两所连续的房子不被抢,例如:
Input:
[2, 1, 1, 2]
抢劫房子1和房子4:2 + 2 = 4
故而,每一个状态i只能由第i-2或第i-3个状态得到(若由第i-4个状态得到,加上第i-2个一定会更大)
前面的状态只需预处理一下就好了
代码如下:
class Solution {
public:
int rob(vector<int>& nums) {
int sum = 0;
if(nums.size()==0) return 0;
if(nums.size()==1) return nums[0];
if(nums.size()==2) return max(nums[0], nums[1]);
int dp[nums.size()+1];
dp[0] = nums[0];
dp[1] = nums[1];
dp[2] = nums[2]+nums[0];
for(int i=3; i<nums.size(); i++)
{
dp[i] = nums[i] + max(dp[i-2], dp[i-3]);
}
sum = max(dp[nums.size()-1],dp[nums.size()-2]);
return sum;
}
};