Leetcode 534打劫房屋II python

描述

在上次打劫完一条街道之后，窃贼又发现了一个新的可以打劫的地方，但这次所有的房子围成了一个圈，这就意味着第一间房子和最后一间房子是挨着的。每个房子都存放着特定金额的钱。你面临的唯一约束条件是：相邻的房子装着相互联系的防盗系统，且 当相邻的两个房子同一天被打劫时，该系统会自动报警。

给定一个非负整数列表，表示每个房子中存放的钱， 算一算，如果今晚去打劫，在不触动报警装置的情况下, 你最多可以得到多少钱 。

这题是House Robber的扩展，只不过是由直线变成了圈

您在真实的面试中是否遇到过这个题？
样例
样例1

输入: nums = [3,6,4]
输出: 6
样例2

输入: nums = [2,3,2,3]
输出: 6

思路：

动态规划求解
（1）不考虑第一间房子和最后一间房子是挨着的时
考虑前i项的结果dp[i]时，

dp[i]为到达第i个房间时，得到的最大收益，A为房间钱数组。

当i = 1， 返回dp[0] = A[0]

当i = 2， 返回dp[1] = max(A[0], A[1])

当i = 3， 分为偷3号房屋和不偷3号房屋，

偷的情况下， 2号房间就不能偷了，结果为A[2] + dp[0]

不偷的情况下，结果为dp[1]

所以返回dp[2] = max(dp[0] + A[2], dp[1])

以此类推，dp[i] = max(dp[i-2] + A[i], dp[i-1])
（2）考虑第一间房子和最后一间房子是挨着时

区别在于1号房屋和最后一号房屋只能二选一
即把1号房间舍弃，或者最后一号房间舍弃。在这两种情况下选最优。

class Solution:
    """
    @param nums: An array of non-negative integers.
    @return: The maximum amount of money you can rob tonight
    """

    def houseRobber(self, nums):
        # write your code here
        if len(nums) == 1:
            return nums[0]
        return max(self.houseRobber2(nums[1:]), self.houseRobber2(nums[:-1]))

    def houseRobber2(self, nums):
        if len(nums) == 0:
            return 0
        dp = [0 for _ in nums]
        dp[0] = nums[0]
        for i in range(1, len(nums)):
            if i == 1:
                dp[i] = max(nums[0], nums[i])
            else:
                dp[i] = max(dp[i - 2] + nums[i], dp[i - 1])
        return dp[-1]

c=Solution()
d=c.houseRobber([2,3,2,3])
print(d)

结果：6