LeetCode 打家劫舍(C++)

直接用递归的话，由于递归的效率低，运行时间会不符合要求，所以用下面的迭代的方法，实际上还是递归的思想，不过用一个变量承载递归结果之后，大大提高了运行效率，代码如下：

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);
    }
};