leetcode 198. House Robber

You are a professional robber planning to rob houses along a street. Each house has a certain amount of money stashed, the only constraint stopping you from robbing each of them is that adjacent houses have security system connected and it will automatically contact the police if two adjacent houses were broken into on the same night.

Given a list of non-negative integers representing the amount of money of each house, determine the maximum amount of money you can rob tonight without alerting the police.

Example 1:

Input: [1,2,3,1]
Output: 4
Explanation: Rob house 1 (money = 1) and then rob house 3 (money = 3).
             Total amount you can rob = 1 + 3 = 4.

Example 2:

Input: [2,7,9,3,1]
Output: 12
Explanation: Rob house 1 (money = 2), rob house 3 (money = 9) and rob house 5 (money = 1).
             Total amount you can rob = 2 + 9 + 1 = 12.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        # dp
        n = len(nums)
        if n == 0: return 0
        if n == 1: return nums[0]
        dp = [0]*(n+1)
        dp[0], dp[1] = 0, nums[0]       
        for i in xrange(2, n+1):
            dp[i] = max(dp[i-2] + nums[i-1], dp[i-1])
        return dp[n]

 or O(1) space.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution(object):
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        # dp
        if not nums: return 0
        if len(nums) == 1: return nums[0]
        a = nums[0]
        b = max(nums[0], nums[1])
        for i in xrange(2, len(nums)):
            b, a = max(a+nums[i], b), b
        return b

 精简版:

1
2
3
4
5
6
class Solution:
     
    def rob(self, nums):       
        last, now = 0, 0       
        for i in nums: last, now = now, max(last + i, now)               
        return now

 

posted @   bonelee  阅读(225)  评论(0编辑  收藏  举报
编辑推荐:
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· DeepSeek 开源周回顾「GitHub 热点速览」
点击右上角即可分享
微信分享提示