Python学习第九周第二次作业——LeetCode练习

第九周课后作业，LeetCode上的题目选一道写一下

题目链接：https://leetcode-cn.com/problems/house-robber-ii/description/

第213题 打家劫舍 II

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下，能够偷窃到的最高金额。

示例 1:
输入: [2,3,2]
输出: 3

解释: 你不能先偷窃 1 号房屋(金额 = 2) ，然后偷窃 3号房屋 (金额 = 2), 因为他们是相邻的。

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

先考虑简化的问题，即第一个和最后一个房子不相邻的情况，就相当于在一列数组中取出一个或多个不相邻数，使其和最大。对于这类求极值的问题首先考虑动态规划来解，维护一个数组maxmoney，其中maxmoney[i]表示到i位置时不相邻数能形成的最大和。先拿一个简单的例子来分析一下，比如说nums为{3, 2, 1, 5}，那么maxmoney数组应该是什么样的，首先maxmoney[0]=3没啥疑问，再看maxmoney[1]是多少呢，由于3比2大，所以抢第一个房子的3，当前房子的2不抢，所以maxmoney[1]=3，那么再来看maxmoney[2]，由于不能抢相邻的，所以可以用再前面的一个的maxmoney值加上当前的房间值，和当前房间的前面一个maxmoney值比较，取较大值当做当前maxmoney值，可以得到递推公式maxmoney[i] = max(num[i] + maxmoney[i - 2], maxmoney[i - 1]), 由此看出我们需要初始化maxmoney[0]和maxmoney[1]，其中maxmoney[0]即为num[0]，maxmoney[1]应该为max(num[0], num[1])。


而当第一个和最后一个相邻时，只需要计算除去第一个后可偷窃的最大金额，和除去最后一个个后可偷窃的最大金额，再取两个中较大的，就得到了问题的解。

代码如下

class Solution:
    def rob_initial(self, nums, left, right):
        maxmoney = []
        if left == 1:
            maxmoney.append(0)
        maxmoney.insert(left, nums[left])
        maxmoney.insert(left+1, max(nums[left], nums[left+1]))
        for i in range(left+2, right):
            maxmoney.insert(i, max(maxmoney[i-1], maxmoney[i-2]+nums[i]))  
        return maxmoney[right-1]
    
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums) == 0:
            return 0
        if len(nums) == 1:
            return nums[0]
        if len(nums) == 2:
            return max(nums[0], nums[1])
        return max(self.rob_initial(nums, 0, len(nums)-1), self.rob_initial(nums, 1, len(nums)))