直接用递归的话,由于递归的效率低,运行时间会不符合要求,所以用下面的迭代的方法,实际上还是递归的思想,不过用一个变量承载递归结果之后,大大提高了运行效率,代码如下:
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)
{
if (nums[0] > nums[1])
return nums[0];
else
return nums[1];
}
int first = 0; //类似rob(n-2)的效果
int second = 0; //类似rob(n-1)的效果
int tmp = 0;
for (int i = 0; i < nums.size(); i++)
{
if (first + nums[i] > second)
{
tmp = second;
second = first + nums[i];
first = max(first, tmp);
}
else
{
first = second;
}
}
return max(first, second);
}
};