很明显是dp问题。用一个一维数组表示dp数组,状态转移方程为:
resmax[index] = max(nums[index] + resmax[index-2], resmax[index-1]); 其中resmax保存,遍历到该节点时的,最大值。注意,遍历到该节点,但该节点不一定选。
另外该题的初始解要想清楚。即resmax[0]=nums[0];但是resmax[1]不等于nums[1],而是resmax[1]=max(nums[0],nums[1])。
因为要卡resmax的定义。
class Solution
{
public:
int rob(vector<int>& nums)
{
if(nums.size() == 0)
return 0;
if(nums.size() == 1)
return nums[0];
if(nums.size() == 2)
return max(nums[0], nums[1]);
vector<int> resmax(nums.size()+1, 0);
//基准情况
resmax[0] = nums[0];
resmax[1] = max(nums[1],nums[0]);
for(int index=2;index < nums.size();++index)
{
resmax[index] = max(nums[index] + resmax[index-2], resmax[index-1]);
}
return resmax[nums.size()-1];
}
};