LeetCode刷题笔记第198题:打家劫舍

LeetCode刷题笔记第198题:打家劫舍

想法:
不能劫相邻的房间,并得到最大的打劫金额。设置两个变量first和second,first代表打劫前n-1个房间的最大金额,second代表打劫n个房间的打劫金额。当只有一个房间时最大金额即为这一家的金额,当房间为两个或两个以上时,从头开始遍历房间,判断当前位置所能打劫到的最大金额,循环结束则返回打劫的最大金额。
使用了动态规划的思想:

  1. 边界:
    d p [ 0 ] = n u m s [ 0 ] dp[0]=nums[0] dp[0]=nums[0]只有一个房间,打劫最大金额为此房间的金额
    d p [ 1 ] = m a x ( d p [ 0 ] , d p [ 1 ] ) dp[1]=max(dp[0], dp[1]) dp[1]=max(dp[0],dp[1])有两个房间,打劫最大金额为两个房间中的值最大的那个
  2. 最优子结构:
    要求打劫n个房间的最大金额就要先求打劫n-1个房间的最大金额
  3. 状态转移方程
    d p [ n ] = m a x ( d p [ n − 2 ] + n u m s [ n ] , d p [ n − 1 ] ) dp[n]=max(dp[n-2]+nums[n], dp[n-1]) dp[n]=max(dp[n2]+nums[n],dp[n1])
    打劫n个房间的最大值为打劫n-1个房间与打劫n个房间中较大的值
class Solution:
    def rob(self, nums: List[int]) -> int:
        # 当列表不存在时
        if not nums:
            return 0
        # 当列表中只有一个元素
        if len(nums) == 1:
            return nums[0]
        # 滚动赋值,first代表的是前n-1个的最大值,second代表的是前n个的最大值
        first, second = nums[0], max(nums[0], nums[1])
        for i in range(2, len(nums)):
            first, second = second, max(first+nums[i], second)

        return second
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值