(一)很基础的动态规划。因为不能取连续的,所以要么只取第i-1个或者取第i-2和第i个,去最大值,列出状态转移方程即可。
Leetcode:198
class Solution {
public:
int rob(vector<int>& nums) {
int n=nums.size();
if(n==0)//需要特判,否则为0时越界
return 0;
else{
vector<int> dp(n,0);
dp[0]=nums[0];
dp[1]=max(nums[0],nums[1]);
for(int i=2;i<nums.size();i++)
{
dp[i]=max(dp[i-1],dp[i-2]+nums[i]);
}
return dp[nums.size()-1];
}
}
};
(二) Leetcode 213
是上面的题目的升级版,意思是这几个House围成一个圈,不能连续抢劫相邻的两个,究竟变得地方是哪里?
显然:第一个和最后一个可能会有冲突,所以,我们由0~n-2进行一次dp,1~n-1进行一次dp,两者取最大的就可以了,这样可以巧妙地避开了首尾可能都取的错误,核心代码和思想和上题一致。
class Solution {
public:
int rob(vector<int>& nums) {
int ans;
int n=nums.size();
if(n==0)
return 0;
else
{
vector<int> dp(n,0);
dp[0]=nums[0];
dp[1]=max(nums[0],nums[1]);
for(int i=2;i<n-1;i++)
dp[i]=max(dp[i-1],dp[i-2]+nums[i]);
ans=dp[n-2];
dp[1]=nums[1];
dp[2]=max(nums[1],nums[2]);
for(int i=3;i<n;i++)
dp[i]=max(dp[i-1],dp[i-2]+nums[i]);
ans=max(ans,dp[n-1]);
return ans;
}
}
};