动态规划
198.House Robber(Easy)
题目描述:
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
思路:
定义一个数组dp,dp[i]表示打劫到第i个屋子时,最大的物资。
如果抢劫第i个屋子,则不抢劫第i-1个屋子,当前最大的抢劫数量为dp[i-2]+nums[i];
如果不抢劫第i个屋子,则抢劫第i-1个屋子,当前最大的抢劫数量为dp[i-1];
状态转移方程:dp[i] = max(dp[i-1],nums[i] + dp[i-2])
初始边界条件为i等于0和1,要保证状态方程有意义。
代码:
class Solution:
def rob(self, nums: List[int]) -> int:
n = len(nums)
'''注意一定要定义dp,一开始我就忘了,报错没有定义dp'''
dp = [0] * n
if n == 0: return 0
if n == 1: return nums[0]
'''初始化边界条件'''
dp[0] = nums[0]
dp[1] = max(nums[0],nums[1])
'''循环条件'''
for i in range(2, n):
'''状态转移方程'''
dp[i] = max(dp[i-1],nums[i] + dp[i-2])
return dp[n-1]