leetcode: 打家劫舍(动态规划 + 滚动数组)

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。

一刷:

用一个数组来存放前i个房屋最多能偷盗的金额,只要单独考虑第1,2和大于2的情况

class Solution:
    def rob(self, nums: List[int]) -> int:
        i = 0
        n = len(nums)
        if n == 0:
            return 0
        if n == 1:
            return nums[0]
        dp = [0 for x in range(n)]
        dp[0] = nums[0]
        dp[1] = max(nums[0], nums[1])
        for i in range(2, n):
            dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
        return dp[n-1]

其实不需要数组来存放,只需要两个记录前两间房最大金额就可以了

class Solution:
    def rob(self, nums: List[int]) -> int:
        i = 0
        n = len(nums)
        if n == 0:
            return 0
        if n == 1:
            return nums[0]
        dp = [0 for x in range(n)]
        first = nums[0]
        second = max(nums[0], nums[1])
        for i in range(2, n):
             tmp = max(second, first + nums[i])
             first = second
             second = tmp
        return  second

 

 

 

你可能感兴趣的:(python)