題目:
你是一個專業的小偷,計劃偷竊沿街的房屋,每間房內都藏有一定的現金。這個地方所有的房屋都圍成一圈,這意味着第一個房屋和最後一個房屋是緊挨着的。同時,相鄰的房屋裝有相互連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警。
給定一個代表每個房屋存放金額的非負整數數組,計算你在不觸動警報裝置的情況下,能夠偷竊到的最高金額。
#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;
}
這個例子就是拿第一個房間,不拿最後一個房間的物品。