LeetCode213. House Robber II(python)

本文探讨了一种特殊情况下房屋抢劫问题的解决方案,即房屋排列成环形,且相邻房屋有安全系统相连,若同时被抢则会报警。通过将问题分解为两部分并使用动态规划算法,最终找到在不触动警报的情况下,能抢到的最大金额。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

You are a professional robber planning to rob houses along a street. Each house has a certain amount of money stashed. All houses at this place are arranged in a circle. That means the first house is the neighbor of the last one. Meanwhile, adjacent houses have security system connected and it will automatically contact the police if two adjacent houses were broken into on the same night.

Given a list of non-negative integers representing the amount of money of each house, determine the maximum amount of money you can rob tonight without alerting the police.

Example 1:

Input: [2,3,2]
Output: 3
Explanation: You cannot rob house 1 (money = 2) and then rob house 3 (money = 2),
             because they are adjacent houses.

Example 2:

Input: [1,2,3,1]
Output: 4
Explanation: Rob house 1 (money = 1) and then rob house 3 (money = 3).
             Total amount you can rob = 1 + 3 = 4.

这是一道动态规划算法题,最开始的想法就是dp[i]=dp[i]+max(dp[i-2],dp[i-3]),但是问题在于这是一个首位循环的题,第一个和最后一个不能重复计算,需要考虑在内。

所以我把这个list分成了两部分,一部分是[1:]另一部分是[:-1]也就是一个不算首,一个不算尾,然后再取两者最大值即可。

class Solution:
    def rob(self, nums: List[int]) -> int:

        if len(nums)<3:return max([0]+nums)

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

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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值