题目描述
抢劫一排住户,不能抢相邻的房屋,求最大抢劫量。
非负整数数组代表每个房屋存放的金额
思路
动态规划
定义 dp 数组用来存储最大抢劫量,其中 dp[i] 表示抢到第 i 个住户时的最大抢劫量。
由于不能抢相近的住户,那么抢了第 i - 1 个住户,就不能再抢第 i 个住户。
在当前位置第 i 个住户 的最大抢劫量,要么为第 i - 1 个住户的最大抢劫量,要么为第 i - 2 个住户的最大抢劫量加上当前房屋的值nums[i - 1],二者之间取最大值。
所以动态规划方程为:
dp[i] = Math.max(nums[i - 1] + dp[i - 2], dp[i - 1])
时间复杂度:O(n)
图解参考:https://leetcode-cn.com/problems/house-robber/solution/hua-jie-suan-fa-198-da-jia-jie-she-by-guanpengchn/
代码
注意
- dp数组的下标从 0 开始,所以dp数组长度为 n + 1
- 第 i 个房屋对应的当前房屋的金额为nums[i - 1]
- dp[2] 需要借助 dp[0]
- 遍历nums,i 从 2 开始,到 n 结束 (i <= n)
- 若使用两个变量 pre1 和 pre2 交替记录 dp[i - 1] 与 dp[i - 2],那么( i = 0;i < n),第 i 个房屋对应的当前房屋的金额为nums[i]