198 打家劫舍
这问题并不是只有隔一个抢一个两种方式,因为可以隔很多个抢一个实现最大收益
背包问题,容量最大是所有家庭的金钱总和(不可能超过,最多相等)
物品是家庭,但是禁止连续拿相邻的两个物品
dp表示能抢到的最大金额
每家只能抢一次,01背包,无所谓顺序,先遍历物品再遍历背包,且内层循环从大到小
写不出来,陷入背包思维定式了
动态规划的思想就是根据前一个状态递归当前状态
考虑影响当前dp的关键因素:是否偷当前的房子
只跟前一个和前两个dp有关
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums) == 1: return nums[0]
dp = [0]*len(nums)
dp[0] = nums[0]
dp[1] = max(nums[:2])
for i in range(2,len(nums)):
dp[i] = max(dp[i-2]+nums[i],dp[i-1])
return dp[-1]
213 打家劫舍2
与原题只有一处变化,不允许同时抢劫第一个和最后一个房子
改变初始化策略,从最大的一个开始遍历,如此必然不会偷最后一个
提交发现思路有误,不能包含所有的情况
看了解答,居然是最简单粗暴的跑两遍比大小
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums)==1: return nums[0]
res1 = self.dp(nums[:-1])
res2 = self.dp(nums[1:])
# print(res1,res2)
return max(res1,res2)
def dp(self,nums):
if len(nums) == 1: return nums[0]
dp = [0]*len(nums)
dp[0] = nums[0]
dp[1] = max(nums[:2])
for i in range(2,len(nums)):
dp[i] = max(dp[i-1],dp[i-2]+nums[i])
return dp[-1]
还是得注意输入的特殊情况!!
337 打家劫舍3
树形dp,结合了二叉树的遍历和dp递推公式
先确定一下用什么遍历方式?
根节点 = max(左孩子+右孩子,自身)
后序遍历?
没写出来,看了解答发现每个dp元素需要存储两个数值,一个偷当前节点一个不偷当前节点的最大金额
class Solution:
def rob(self, root: Optional[TreeNode]) -> int:
return max(self.recur(root))
def recur(self, root):
if root == None:
return [0,0]
left = self.recur(root.left)
right = self.recur(root.right)
return [left[1]+right[1]+root.val,max(left)+max(right)]
还有一个点是,当不偷根节点时,左右孩子可偷,但是具体偷不偷得取最大的,不能无脑选左右孩子都偷