题目
思考
刚开始想的dp[i]的解释是以nums[i]为结尾(包括nums[i])的最大数量,然后找不到状态方程。(这道题要和最长的子序列和区别开来,最长子序列和是连续的,故需要以dp[i]为结尾,而这道题可以是跳跃的,故不需要以dp[i]为结尾)
而把dp[i]理解为到位置i位置所能偷的最大数量,就能很容易的写出状态方程。
代码
class Solution {
public:
int rob(vector<int>& nums) {
if(nums.size()==0) return 0;
if(nums.size()==1) return nums[0];
//从第一个开始
vector<int> dp1(nums.size(),0);
//从第二个开始
vector<int> dp2(nums.size(),0);
dp1[1]=nums[0];
for(int i=2;i<=nums.size()-1;i++)
dp1[i]=max(dp1[i-1],dp1[i-2]+nums[i-1]);
dp2[1]=nums[1];
for(int i=2;i<=nums.size()-1;i++)
dp2[i]=max(dp2[i-1],dp2[i-2]+nums[i]);
return max(dp1[nums.size()-1],dp2[nums.size()-1]);
}
};