题目描述
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.
假如你是一个强盗–当然是一个希望尽可能抢到更多的钱的强盗. 现在nums数列给出了一排房子, 数列下标i对应的值表示了第i间房子有多少钱. 但要注意到, 一个强盗他必定不能引起警察的注意, 而如果有连续的房子被抢, 就会引起警报招来警察.
解题思路
这个问题实际是就是一个简单的动态规划的变种–变就变在我们需要保存两个状态.
这道题Bellman Equation为:
当前i能抢到的最多金额=max(抢第i间房子所能得到的最大金额,不抢第i间房子所能得到的最大金额)
当
前
i
能
抢
到
的
最
多
金
额
=
m
a
x
(
抢
第
i
间
房
子
所
能
得
到
的
最
大
金
额
,
不
抢
第
i
间
房
子
所
能
得
到
的
最
大
金
额
)
抢第i间房子的最大金额=不抢第i−1间房子的最大金额+第i间房子的金额
抢
第
i
间
房
子
的
最
大
金
额
=
不
抢
第
i
−
1
间
房
子
的
最
大
金
额
+
第
i
间
房
子
的
金
额
不抢第i间房子的最大金额=max(抢第i−1间房子的最大金额,不抢第i−1间房子的最大金额)
不
抢
第
i
间
房
子
的
最
大
金
额
=
m
a
x
(
抢
第
i
−
1
间
房
子
的
最
大
金
额
,
不
抢
第
i
−
1
间
房
子
的
最
大
金
额
)
根据以上3个式子写出代码即可快速求解.
代码实现
下面的代码是我第一次AC的代码. 但其实它还有挺大的优化空间–其实不必要保存所有的房子抢或不抢能得到的最大金额, 也就是说代码所开的二维数组时不必要的. 我的代码后面附上的即是这道题上运行时间较短的另一种解法.
class Solution(object):
def rob(self, nums):
if len(nums) == 0:
return 0
cache = [[0 for i in range(len(nums))] for i in range(2)]
cache[1][0] = nums[0]
for i in range(1, len(nums)):
cache[0][i] = max(cache[0][i - 1], cache[1][i - 1])
cache[1][i] = cache[0][i - 1] + nums[i]
# print(cache)
return max(cache[0][len(nums) - 1], cache[1][len(nums) - 1])
# Better Solution
class Solution(object):
def rob(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
pre, cur = 0, 0
for n in nums:
pre, cur = cur, max(pre+n, cur)
return cur