动态规划---打家劫舍2

mac2022-06-30  25

 题目:

你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。

#include <iostream> #include <algorithm> #include <string> #include <vector> #include <stack> #include <map> #include <queue> using namespace std; //打家劫舍II 环形房间 //要么不拿第一个房间的东西,要么不拿最后一个房间的东西,价值对比,取最终最大价值 #if 1 class Solution { public: int rob(vector<int>& nums) { int len = nums.size(); if(len == 0) return 0; if(len == 1) return nums[0]; if(len == 2) return nums[0] < nums[1] ? nums[1] : nums[0]; return max(findmax(nums,0,len-2),findmax(nums,1,len-1)); } private: int findmax(vector<int>& nums,int start,int end) { int len = nums.size(); vector<int> dp(len); dp[0] = nums[start]; dp[1] = nums[start] < nums[start+1] ? nums[start+1] : nums[start]; for(int i = 2+start;i <= end;++i) { dp[i-start] = max(dp[i-start-2]+nums[i],dp[i-1-start]); } return dp[end-start]; } }; int main() { int n; cin >> n; vector<int> vec(n); for(int i = 0;i < n;i++) { cin >> vec[i]; } Solution s; cout << s.rob(vec) << endl; }

这个例子就是拿第一个房间,不拿最后一个房间的物品。

最新回复(0)