打卡群刷题总结0920——打家劫舍 II

时间:2022-07-26
本文章向大家介绍打卡群刷题总结0920——打家劫舍 II,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

题目:213. 打家劫舍 II

链接:https://leetcode-cn.com/problems/house-robber-ii

你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。 给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。 示例 1: 输入: [2,3,2] 输出: 3 解释: 你不能先偷窃 1 号房屋(金额 = 2),然后偷窃 3 号房屋(金额 = 2), 因为他们是相邻的。

解题:

1、dp问题。分为两种情况:偷第一家(不能偷最后一家)和不偷第一家(可以偷最后一家)。

dp公式和昨天一样:dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])

代码:

class Solution:
    def rob(self, nums: List[int]) -> int:
        if len(nums) == 0:
            return 0
        res = 0
        # 偷第一家
        pre, cur = 0, nums[0]
        for i in range(1, len(nums) - 1):
            pre, cur = cur, max(cur, pre + nums[i])
        res = max(pre, cur)
        
        # 不偷第一家
        pre, cur = 0, 0
        for i in range(1, len(nums)):
            pre, cur = cur, max(cur, pre + nums[i])
        res = max(res, pre, cur)
        return res