1. 程式人生 > >26.動態規劃-打家劫舍-Leetcode 198(python)

26.動態規劃-打家劫舍-Leetcode 198(python)

  • 題目描述

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

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

  • 示例

 

示例 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 。

  • 解決思路

動態規劃思路,按照不能打劫相鄰家庭的思路,建立另一個數組dp用來存放走到第i戶人家時,最多可以打劫到的金額。有dp[i] = max(dp[i-1],dp[i-2]+nums[i]).

學到python建立陣列的方法 dp = [0]*len(nums)

  • 程式碼
class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums) == 0:
            return 0
        if len(nums) == 1:
            return nums[0]
        if len(nums) == 2:
            return max(nums[0],nums[1])
        
        #固定長度的陣列的定義
        dp = [0] *len(nums)
        dp [0] = nums[0]
        dp [1] = max(nums[0],nums[1])

        for i in range(2,len(nums)):
            dp[i] = max(dp[i-1],dp[i-2] + nums[i])

        return dp[len(nums)-1]