動態規劃---打家劫舍2

 題目:

你是一個專業的小偷,計劃偷竊沿街的房屋,每間房內都藏有一定的現金。這個地方所有的房屋都圍成一圈,這意味着第一個房屋和最後一個房屋是緊挨着的。同時,相鄰的房屋裝有相互連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警。

給定一個代表每個房屋存放金額的非負整數數組,計算你在不觸動警報裝置的情況下,能夠偷竊到的最高金額。

#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;
}

這個例子就是拿第一個房間,不拿最後一個房間的物品。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章