【LeetCode 213】打家劫舍II Python 精简代码
【解题思路后续补上】
直接看代码
class Solution:
def rob(self, nums: List[int]) -> int:
""" 首先,首尾房间不能同时被抢,那么只可能有三种不同情况:要么都不被抢;要么第一间房子被抢最后一间不抢;要么最后一间房子被抢第一间不抢。 那就简单了啊,这三种情况,那种的结果最大,就是最终答案呗!不过,其实我们不需要比较三种情况,只要比较情况二和情况三就行了,因为这两种情况对于房子的选择余地比情况一大呀,房子里的钱数都是非负数,所以选择余地大,最优决策结果肯定不会小。 【参考如下】 https://leetcode-cn.com/problems/house-robber-ii/solution/tong-yong-si-lu-tuan-mie-da-jia-jie-she-wen-ti-by-/ https://leetcode-cn.com/problems/house-robber-ii/solution/yi-ge-forxun-huan-jie-jue-wen-ti-by-mu-tu-ze-shuai/ """
n = len(nums)
if n == 0:
return 0
if n == 1:
return nums[0]
# 如果写成 dp1 = dp2 = [0] * (n+1)会出错,这样dp1和dp2两个指针都指向列表[0]*(n+1),改变一个,另一个也改# 变,但是我们要的是两个独立的列表,只有赋常数的时候可以这样赋值,列表或者字典不能连续赋值
dp1 = [0] * (n+1)
dp2 = [0] * (n+1)
for i in range(2, n+1):
dp1[i] = max(dp1[i-2] + nums[i-2], dp1[i-1])
dp2[i] = max(dp2[i-2] + nums[i-1], dp2[i-1])
return max(dp1[n], dp2[n])
####################################################################3
""" index = 0, 1, 2, 3, ..., n-1, n dp1 = [0, #, #, #, ..., #, #] len(dp1) = n+1 -------------------------------------------------- index = 0, 1, 2, 3, ..., n-1, n dp2 = [0, #, #, #, ..., #, #] len(dp2) = n+1 ------------------------------------------------------ index = 0, 1, 2, 3, ..., n-1 nums = [#, #, #, #, ..., #] len(nums) = n [#, #, #, ..., #] 去尾 nums[0:n-2] [#, #, #, ..., #] 去头 nums[1:n-1] """
#####################################################################