其他分享
首页 > 其他分享> > 动态规划——打家劫舍(力扣198:打家劫舍)

动态规划——打家劫舍(力扣198:打家劫舍)

作者:互联网

题目描述

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额

输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
     偷窃到的最高金额 = 1 + 3 = 4 。

输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
     偷窃到的最高金额 = 2 + 9 + 1 = 12 。

题目链接:198. 打家劫舍 - 力扣(LeetCode) (leetcode-cn.com)

分析

如何想到用动态规划解题

我们就以第二个例子来分析, 一共五户人家, 考虑偷到第三家的情况。 要在第三家偷到的钱财最多, 那么就要看第一第二家的选择, 因为不能连续偷挨着的两家, 那么实际上就是要比较第二家的钱财与第一家+第三家的钱财哪个更多的问题。 而与后面怎么偷毫无关系(只考虑偷到第三家的时候), 像这种题, 某一个状态只与前面的状态有关,与后面的状态无关的情况, 一半考虑用动态规划解题

动态规划解题

代码

class Solution:
    def rob(self, nums):
        N = len(nums)
        if N > 100 or N < 1:
            raise ValueError("Invalid Input")

        if N == 1:
            return nums[0]

        if N == 2:
            return max(nums[0], nums[1])

        value_front, value_end = nums[0], max(nums[0], nums[1])
        for i in range(2, N):
            temp = max(value_end,  value_front + nums[i])
            value_front = value_end
            value_end = temp

        return value_end

标签:偷窃,198,nums,max,value,力扣,end,打家劫舍,dp
来源: https://blog.csdn.net/qq_40472852/article/details/121741879