day52 动态规划part9 代码随想录算法训练营 213. 打家劫舍 II
题目:213. 打家劫舍 II
我的感悟:
- 看了题解不难,就是环这个思路转化很重要!
理解难点:
- 环的转化为,首,尾。
- 代码上面可以省略长度为2的校验
听课笔记:
- 分3中情况:
- 不考虑首尾 | 考虑首 | 考虑尾
- 而情况2和情况3 包含了情况1
我的代码:
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums) == 1:
return nums[0]
# if len(nums) == 2:
# return max(nums[0],nums[1])
# 情况2:只包含首元素
res1 = self.rob_base(nums[:len(nums)-1])
# 情况3:只包含尾元素
res2 = self.rob_base(nums[1:])
return max(res1,res2)
def rob_base(self,nums):
if len(nums) == 0:
return 0
if len(nums) == 1:
return nums[0]
dp = [0] *len(nums)
dp[0] = nums[0]
dp[1] = max(nums[0],nums[1])
for i in range(2,len(nums)):
dp[i] = max(dp[i-1],nums[i]+dp[i-2])
# print(dp)
return dp[-1]
通过截图:
老师代码:
python网友没我写的好
扩展写法:
资料:
213.打家劫舍II
视频讲解:https://www.bilibili.com/video/BV1oM411B7xq
https://programmercarl.com/0213.%E6%89%93%E5%AE%B6%E5%8A%AB%E8%88%8DII.html