1. 程式人生 > 實用技巧 >TS 3.1 - 型別推論

TS 3.1 - 型別推論

技術標籤:leetcodepython

“”"
你是一個專業的小偷,計劃偷竊沿街的房屋。
每間房內都藏有一定的現金,影響你偷竊的唯一制約因素就是相鄰的房屋裝有相互連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警。
給定一個代表每個房屋存放金額的非負整數陣列,計算你 不觸動警報裝置的情況下 ,一夜之內能夠偷竊到的最高金額。
示例 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 。
“”"

def rob(nums):
    if not nums:
        return 0

    n = len(nums)
    if n == 1:
        return nums[0]

    dp = [0 for _ in range(n)]
    # 只有一間房屋,則偷竊該房屋
    dp[0] = nums[0]
    # 只有兩間房屋,選擇其中金額較高的房屋進行偷竊
    dp[1] = max(nums[0], nums[1])
    # 偷竊第 k 間房屋,那麼就不能偷竊第 k−1 間房屋,偷竊總金額為前 k−2 間房屋的最高總金額與第 k 間房屋的金額之和。
    # 不偷竊第 k 間房屋,偷竊總金額為前 k−1 間房屋的最高總金額。
    for i in range(2, n):
        dp[i] = max(dp[i - 2] + nums[i], dp[i - 1])

    return dp[-1]


if __name__ == '__main__':
    nums = [1, 2, 3, 1]
    print(rob(nums))