leetcode - 213. 打家劫舍 II

最新推荐文章于 2024-07-18 01:42:44 发布

zmm_mohua

最新推荐文章于 2024-07-18 01:42:44 发布

阅读量265

点赞数

分类专栏：刷题文章标签： leetcode c++

本文链接：https://blog.csdn.net/qq_42250642/article/details/125745929

版权

刷题专栏收录该内容

312 篇文章 0 订阅

订阅专栏

leetcode - 213. 打家劫舍 II

题目

在这里插入图片描述

代码

#include <iostream>
#include <vector>
using namespace std;

int robMoney(vector<int>& nums, int start, int end){
	int first = nums[start], second = max(nums[start], nums[start+1]);
	for(int i = start + 2; i <= end; i++){
		int temp = second;
		second = max(nums[i] + first, second);
		first = temp;
	}
	return second;
}

int rob(vector<int>& nums) {
	int n = nums.size();
	if(n == 1){
		return nums[0];
	}else if(n == 2){
		return max(nums[0], nums[1]);
	}
	return max(robMoney(nums, 0, n-2), robMoney(nums, 1, n-1));
}

int main(){
	int n, res;
	cin>>n;
	vector<int> nums(n);
	for(int i = 0; i < n; i++){
		cin>>nums[i];
	}
	res = rob(nums);
	cout<<res;
    return 0;
}