1. 程式人生 > >LeetCode 198.House Robber (打家劫舍)

LeetCode 198.House Robber (打家劫舍)

題目描述:

你是一個專業的小偷,計劃偷竊沿街的房屋。每間房內都藏有一定的現金,影響你偷竊的唯一制約因素就是相鄰的房屋裝有相互連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警

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

示例 1:

輸入: [1,2,3,1]
輸出: 4
解釋: 偷竊 1 號房屋 (金額 = 1) ,然後偷竊 3 號房屋 (金額 = 3)。
     偷竊到的最高金額 = 1 + 3 = 4 。

示例 2:

輸入: [2,7,9,3,1]
輸出: 12
解釋: 偷竊 1 號房屋 (金額 = 2), 偷竊 3 號房屋 (金額 = 9),接著偷竊 5 號房屋 (金額 = 1)。
     偷竊到的最高金額 = 2 + 9 + 1 = 12 。

 

AC C++ Solution:

典型的動態規劃題,第i號房屋的最大金額:dp(i) = max(dp[i-2]+nums[i], dp[i-1])

 

class Solution {    
public:
    int rob(vector<int>& nums) {                            //動態規劃
        if (nums.size() <= 1)
            return !nums.empty() ? nums[0] : 0; 
        vector<int> dp{nums[0], max(nums[0], nums[1])};     //初始化
        for (int i = 2; i < nums.size(); ++i)
        {
            dp.push_back(max(nums[i] + dp[i - 2], dp[i-1]));    //第i號的最大金額為max(dp[i-2]+nums[i] , dp[i-1]);
        }
        return dp.back();
    }
};