题解 | #打家劫舍(二)#

打家劫舍(二)

https://www.nowcoder.com/practice/a5c127769dd74a63ada7bff37d9c5815

class Solution:
    def rob(self , nums: List[int]) -> int:
        dp = [0 for _ in range(len(nums))]
        dp[1] = nums[0]
        for i in range(1, len(nums)):
            dp[i] = max(dp[i-1], dp[i-2]+nums[i-1])

        maxNum = dp[-1]

        dp = [0 for _ in range(len(nums)+1)]
        dp[1] = 0
        for i in range(2, len(nums)+1):
            dp[i] = max(dp[i-1], dp[i-2]+nums[i-1])

        
        return max(maxNum, dp[-1])

解题思路

偷第一家的则不偷最后一家的,偷最后一家的则不偷第一家的。对于前面这种情况初始状态为dp[1]=nums[0],遍历时的边界为[1, len(nums)],对于后者则有dp[1]=0, [2, len(nums)+1]。转移状态方程为dp[i] = max(dp[i-1], dp[i-2]+nums[i-1])。最后取两种情况的最大值即可。

复杂度

时间复杂度和空间复杂度都为O(N)。

全部评论

相关推荐

群星之怒:不是哥们,你就不好奇瘫痪三十年的老植物人是啥样的吗?
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务