LeetCode 213 House Robber II Python

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

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

难度:medium

思路:该题在House Robber的基础上让首位链接形成环,那么即表示第一个和最后一个不能同时被抢,则问题分解为House Robber(nums[0:len(nums)-1])和House Robber(nums[1:len(nums)]),两者中比较大的那个即为结果

class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        length = len(nums)
        if length == 0:
            return 0
        elif length == 1:
            return nums[0]
        elif length ==2:
            return max(nums[0], nums[1])
        return max(self.dfs(nums[1:]), self.dfs(nums[:-1]))

    def dfs(self, nums):
        if len(nums) == 2:
            return max(nums[0], nums[1])
        res = [0 for each in nums]
        res[0] = nums[0]
        res[1] = max(nums[0], nums[1])
        for index in xrange(2,len(nums)):
            res[index] = max(res[index -2] + nums[index], res[index - 1])
        return res[len(nums) - 1]

 

你可能感兴趣的:(leetcode,python)