打家劫舍是最为经典的DP题之一,先看题目。
这是什么傻吊包茎系统?两件相邻的房子被偷才会报警?真要这样,写啥LeetCode,找啥工作兄弟们,直接抄家伙偷东西去。
DP题最为关键的一点就是找到DP方程。如果能写出DP方程,这种题目就成功了一大半。
首先我们先设定下DP数组的意义。DP[n]是在我走过第n个房子之后,我能抢到的最大金额(以0为第一个房子)。在示例1中,我们的结果就是DP[3],也就是我们走过第3个,也就是最后1个房子后,能拿到的最大金额。
搞懂DP数组表达的意义后,这题我们可以这样理解,当你经过第n个屋子时,你有两个选择,抢或者不抢。
- 抢这个屋子的话,我们就不能抢上个屋子,如此一来,你的总收益就是DP[n-2]+这个房子里的钱,DP[n-2]的意思就是,你经过上上个房子后,能得到的最多金额。
- 不抢这个屋子的话,你的总收益保持不变,也就是等于DP[n-1]。
我们肯定要选择两者之间,收益最高的那者。所以我们的DP方程也就是DP[n]=max(DP[n-2]+nums[n], DP[n-1])。
代码部分如下。
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size() == 1) return nums[0];
//dp[0]=m
int m = nums[0];
//dp[1]=n
int n = max(nums[0],nums[1]);
for(int i = 2;i < nums.size();++i){
//dp[n]=max(nums[n]+dp[n-2],dp[n-1])
int temp = max(m+nums[i],n);
m = n;
n = temp;
}
return n;
}
};