代码随想录:
将一个线性数组换成两个线性数组(去掉头,去掉尾)
分别求两个线性数组的最大值
最后求这两个数组的最大值
代码随想录视频
#include<iostream>
#include<vector>
#include<algorithm>
//nums:2,1,1,2
//dp:2,2,3,4
using namespace std;
int rop(vector<int> &nums)
{
vector<int>dp(nums.size(), 0);
dp[0] = nums[0];
dp[1] = max(nums[1], nums[0]);
for (int i = 2;i < nums.size();i++)
{
dp[i] = max(dp[i - 1], dp[i - 2] + nums[i]);
}
return dp[nums.size()-1];
}
int main()
{
vector<int>nums = { 1,2,3,1 };
if (nums.size() == 1)
{
cout << nums[0];
return 0;
}
if (nums.size() == 2)
{
cout << max(nums[0],nums[1]);
return 0;
}
vector<int>nums1(nums.begin() + 0, nums.end() - 1);
vector<int>nums2(nums.begin() + 1, nums.end());
cout << max(rop(nums1),rop(nums2));
return 0;
}