LeetCode刷题之198.打家劫舍

LeetCode刷题之198.打家劫舍

我不知道将去向何方,但我已在路上!
时光匆匆,虽未曾谋面,却相遇于斯,实在是莫大的缘分,感谢您的到访 !
  • 题目
    你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。
  • 示例
示例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 。
  • 代码:
class Solution:
    def rob(self, nums: List[int]) -> int:
        n = len(nums)
        if n == 0:
            return 0
        if n <= 2:
            return max(nums)
        i = 3
        result = [nums[0]] + [nums[1]] + [nums[0] + nums[2]] + [0] * (n-3)
        while i < n:
            result[i] = max(result[i-2]+nums[i], result[i-3]+nums[i])
            i += 1
        return max(result)
# 执行用时 :52 ms, 在所有 Python3 提交中击败了56.19%的用户
# 内存消耗 :13.7 MB, 在所有 Python3 提交中击败了5.14%的用户
  • 算法说明:
    首先将nums中个数是0和2的先去除掉;然后判断不少于3的情况;两个房子中间隔三个房子的情况不存在。然后从有三个房间开始讨论:
    (1)假设小偷先进的0号房间,将0号房间的金额,存储在result中;
    (2)假设小偷先进的1号房间,将1号房间的金额,存储在result中;
    (3)由于有三个房间,将第0号房间和第2号房间的金额求和,存入result中,result的其余元素用0进行填充;
    (4)遍历到第i号房间的时候,判断result中的i-2和i-3两个元素与当前元素的和的大小,选较大者存入result中,直至遍历完所有元素。
    (5)返回result中最大的值。
发布了90 篇原创文章 · 获赞 1 · 访问量 1028

猜你喜欢

转载自blog.csdn.net/qq_34331113/article/details/103872116
今日推荐