(leetcode 213 打家劫舍ii)

(leetcode 213 打家劫舍ii)_第1张图片

代码随想录:

将一个线性数组换成两个线性数组(去掉头,去掉尾)

分别求两个线性数组的最大值

最后求这两个数组的最大值

代码随想录视频

(leetcode 213 打家劫舍ii)_第2张图片

#include
#include
#include
//nums:2,1,1,2
//dp:2,2,3,4
using namespace std;

int rop(vector &nums)
{
	
	vectordp(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()
{
	vectornums = { 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;
	}
	vectornums1(nums.begin() + 0, nums.end() - 1);
	vectornums2(nums.begin() + 1, nums.end());


	cout << max(rop(nums1),rop(nums2));
	return 0;
}

你可能感兴趣的:(leetcode,算法,数据结构)