BM79 打家劫舍(二)（python）

最新推荐文章于 2023-08-21 18:02:32 发布

LaLuzEsComoElAgua

最新推荐文章于 2023-08-21 18:02:32 发布

阅读量836

点赞数

分类专栏：算法文章标签：算法 python

本文链接：https://blog.csdn.net/taozhi867243023/article/details/123956367

版权

算法专栏收录该内容

26 篇文章 0 订阅

订阅专栏

本文探讨了一个问题：给定一排环形房间的现金分布，小偷如何选择偷窃路径以获取最大现金，通过动态规划算法求解。文章介绍了问题背景、解决方案思路和Python代码实现，重点在于理解状态转移方程和双指针技巧的应用。

摘要由CSDN通过智能技术生成

题目

题目链接
给出一个长为n的数组nums，表示一排房间中每个房间存放的现金数；
一个小偷沿这排房间偷现金，为防止被发现，不能连续偷相邻两家的现金；所有房间沿湖边排布，形成一个环形，第一个房间与最后一个房间视为相邻；求小偷能够偷得的现金最大值

思路

如果偷了第一家，就不能再偷最后一家
如果没偷第一家，可以偷最后一家

对nums[1:]（不偷第一家）和nums[:-1]（偷第一家）分别进行一次dp，取两次dp的最大值即为最终结果

一次dp的过程：

dp[i]表示到第i个房间时，能偷到的最大值
初始化dp数组为全0，dp[0]=nums[0]，dp[1]=max(nums[0],nums[1])

从i=2开始，对每个dp[i]，计算dp[i]=max(dp[i - 2] + nums[i], dp[i - 1])，表示取【偷这一家】和【不偷这一家】的较大值

代码

class Solution:
    def aRob(self, nums):
        if len(nums) == 1:
            return nums[0]
        dp = [0] * len(nums)
        dp[0]=nums[0]
        dp[1]=max(nums[0],nums[1])
        for i in range(2, len(nums)):
            dp[i] = max(dp[i - 2] + nums[i], dp[i - 1])
        return dp[-1]

    def rob(self, nums):
        if len(nums) == 1:
            return nums[0]
        return max(self.aRob(nums[1:]), self.aRob(nums[:-1]))