Leetcode 213. 打家劫舍 II

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

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

class Solution {
public:
    int rob(vector<int>& nums) {

        int size = nums.size();
        if (size == 0) {
            return 0;
        }
        else if (size == 1){
            return nums[0];
        }
        else if (size == 2) {
            return max(nums[0], nums[1]);
        }
        else if (size == 3) {
            return max(max(nums[0], nums[1]), nums[2]);
        }


        int maxNum = 0;
        vector<int> dp1(size, 0);
        dp1[0] = nums[0];
        dp1[1] = dp1[0];
        for (int i=2; i<size-1; i++) {
            dp1[i] = max(dp1[i-1], dp1[i-2] + nums[i]);
            maxNum = max(maxNum, dp1[i]);
        }

        vector<int> dp2(size, 0);
        dp2[0] = 0;
        dp2[1] = nums[1];
        for (int i=2; i<size; i++) {
            dp2[i] = max(dp2[i-1], dp2[i-2] + nums[i]);
            maxNum = max(maxNum, dp2[i]);
        }

        return maxNum;
    }
};

 

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