LeetCode.198.打家劫舍

2024年6月10日 75次阅读

原题链接:https://leetcode-cn.com/problems/house-robber/
解题思路，首先建立一个长度为nums.length的列表memo。memo的每一位memo[i]表示劫匪从最后一家往前抢，抢到第i家所获得的最大现金。当劫匪到第i家只有两种情况，抢这一家没抢上一家或者不抢这一家抢了上一家。
所以递归式为：
memo[i] = max(nums[i] + memo[i+2] , nums[i+1] + memo[i+3])

class Solution(object):
    def rob(self, nums):
        """ :type nums: List[int] :rtype: int """
        if nums == []:
            return 0
        length = len(nums)
        memo = [0]*length
        memo[-1] = nums[-1]  # 初始条件
        for i in range(length-2,-1,-1):   # 递归式
            memo[i] = max(nums[i] + (0 if i+2>=length else memo[i+2]), \
                          nums[i+1] +(0 if i+3 >= length else memo[i+3]))
        return memo[0]
点赞