一个强盗抢劫n户人家,这n户人家收尾是相连的,抢劫不能抢连续的人家,求抢劫的最大数:
class Solution {
public:
int rob(vector<int>& nums) {
int t=nums.size();
if(t==0)
return 0;
if(t==1)
return nums[0];
//不选择第t个:从第1个到第t-1个
vector<int> dp(t,0);
dp[0]=nums[0];
dp[1]=max(nums[0],nums[1]);
for(int i=2;i<t-1;i++){
dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
}
//选择第t个:从第2个到第t个
vector<int> dp1(t,0);
//第一个不能选
dp1[1]=nums[1];
dp1[2]=max(nums[1],nums[2]);
for(int i=3;i<t;i++){
dp1[i]=max(dp1[i-2]+nums[i],dp1[i-1]);
}
return max(dp[t-2],dp1[t-1]);
}
};