你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
示例 1:
输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
偷窃到的最高金额 = 1 + 3 = 4 。
示例 2:
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
偷窃到的最高金额 = 2 + 9 + 1 = 12 。
提示:
1<=nums.length<=1001 <= nums.length <= 1001<=nums.length<=100
0<=nums[i]<=4000 <= nums[i] <= 4000<=nums[i]<=400
思路:动态规划
- 从左向右分别计算到当前屋子的最高金额,可以发现当前屋子的金额,依赖于道上一个和上上一个屋子的最大金额
- 可以考虑用动态规划解决, f[i] 代表到第 i 个屋子的最大金额
- 对于当前屋子
- 如果在前一个屋子没有盗窃,则最大金额 = 在上上一个屋子盗窃的最大金额 + 当前屋子的最大金额(f[i-2] + nums[i])
- 如果在前一个屋子盗窃了,那当前屋子就不能盗窃(引发警报),则走到当前屋子的最大金额 = 走到上一个屋子的最大金额(f[i-1])
- 状态转移方程:f[i] = max(f[i-1], f[i-2] + nums[i])
- 边界条件为:
- f[0]=nums[0],走到第一个屋子的时候,只能盗窃第一个屋子
- f[1]=max(f[0], nums[1]),走到第二个屋子的时候,只能选择盗窃前一个屋子,或者只盗窃当前屋子(不能同时)
class Solution {
public:
int f[110];
int rob(vector<int>& nums) {
f[0] = nums[0];
int n = nums.size();
for(int i = 1; i < n; i++){
if(i == 1) f[i] = max(f[i-1], nums[i]);
else f[i] = max(f[i-1], f[i-2] + nums[i]);
}
return f[n - 1];
}
};