自己一定要相信自己
分析:
假设只有一家可以打劫,那就偷窃这一家即可。
假设有两家可以偷窃,那就偷窃两者中金额最大的那一家。
假设有三家,那就比较偷窃第一家+第三家的金额之和和第二家的金额做比较,选最大的那个。
设dp[i]是偷窃到当前人家的最大金额数。当前这家有两种选择:
偷
如果偷,那么前一家就不能偷,也就是dp[i-1]不能偷,得到的金额就是dp[i-1]+nums[i]。
不偷
如果不偷,那么前一家就可以偷,得到的金额就是dp[i-1].
由此得到状态转移方程:
dp[i]=max(dp[i-1],dp[i-2]+nums[i])
代码:
class Solution {
public:
int rob(vector<int>& nums) {
//记录数组
int n=nums.size();
int dp[n+1];
//基础值
dp[0]=nums[0];
if(n==1){
return dp[0];
}
dp[1]=max(nums[1],nums[0]);
//公式
//dp[i]=max(dp[i-1],dp[i-2]+nums[i])
for(int i=2;i<n;i++){
dp[i]=max(dp[i-1],dp[i-2]+nums[i]);
}
return dp[n-1];
}
};
在代码里,我做了一些特殊处理,对
- 只有一个数
- 大于等于两个数
分别处理,二者的区别如下:
-
要注意自己的边界是什么,在我的代码里,返回值应该是
dp[n-1]
,因为我的dp[i]
表示打劫到当前人家(如果偷的话,就包括当前人家的金额
)。 -
我看到有的题解是
dp[i]
表示打劫到这家之前的最大金额,不会包括当前这家的金额
,所以最后返回的是dp[n]