下定决心,好好过一天 ~
题目传送
这是简单版本。
思路:
如果拿第 i 个物品,一定可以拿第 i-2 或者 i-3 个物品不然则亏损(没必要),那么动态方程就出来了,直接i从0往后遍历即可。dp[i] = max(nums[i] + dp[i-2] , nums[i] + dp[i-3]),输出dp数组中最大值即可
class Solution {
public:
int rob(vector<int>& nums) {
int dp[1005] = {0};
int len = nums.size();
dp[0] = nums[0];
if(len > 1) dp[1] = nums[1];
if(len > 2) dp[2] = dp[0] + nums[2];
int Max = max({dp[0],dp[1],dp[2]});
for(int i = 3;i < len;i++)
{
int l = i-2,r = i-3;
dp[i] = max(nums[i]+dp[l],nums[i]+dp[r]);
Max = max(Max,dp[i]);
}
// cout << Max << endl;
return Max;
}
};
升级版:
和上面思路差不多,只是,当前是一个环。那么第一个和最后一个不能同时选。
那么就分俩种情况dp,取最大值即可,就是dp从2到n,和从1到n-1。这样肯定满足条件
class Solution {
public:
int rob(vector<int>& nums) {
int dp[1005] = {0};
int len = nums.size();
dp[0] = nums[0];
if(len > 1) dp[1] = nums[1];
if(len > 2 && len != 3) dp[2] = dp[0] + nums[2];
int Max = max({dp[0],dp[1],dp[2]});
for(int i = 3;i < len-1;i++)
{
int l = i-2,r = i-3;
dp[i] = max(nums[i]+dp[l],nums[i]+dp[r]);
Max = max(Max,dp[i]);
}
fill(dp,dp+1005,0);
dp[0] = 0;
if(len > 1) dp[1] = nums[1];
if(len > 2) dp[2] = nums[2];
Max = max({Max,dp[1],dp[2]});
for(int i = 3;i < len;i++)
{
int l = i-2,r = i-3;
dp[i] = max(nums[i]+dp[l],nums[i]+dp[r]);
Max = max(Max,dp[i]);
}
// cout << Max << endl;
return Max;
}
};