leetcode198-打家劫舍

198. 打家劫舍

 

这道题在纸上模拟还是很容易的，但是写代码就不同了。第一次写认为的难点是如何找到当前元素除了上一个之外的最大值。于是我想到了用max1保存第二大的下标，max2保存第二大的下标。

全程写下来还是挺顺的，但是中间有几步dp[i]写成了nums[i]一直没发现就调试了很久。

写的时候没有意识到一些问题。一开始我是直接return dp数组最后一个元素的。但是这有问题

比如数组 1 2 3 1     ,照我写的代码最终dp[i]就是

               1 2 4 3     ，返回最后一个元素明显是错误的，还要比较最后一个元素和max2所指的最大值。

直接把dp方程写为dp[i] = max(dp[i-2]+nums[i], dp[i-1])，那么dp数组中每一个元素都是子元素的答案。而我一开始写这个还有搞各种判断就很麻烦。

class Solution {
public:
    int rob(vector<int>& nums) {
        int size=nums.size();
        if(size==1) return nums[0];
        if(size==2) return max(nums[0],nums[1]);
        vector<int> dp;dp.resize(size);
        dp[0]=nums[0];dp[1]=max(nums[0],nums[1]);
        int max1=0,max2=1;
        for(int i=2;i<size;i++)
        {
            if(max2==i-1)   dp[i]=nums[i]+dp[max1];
            else    dp[i]=nums[i]+dp[max2];
            if(dp[i]>dp[max2])
            {
                max1=max2;
                max2=i;
            }        
        }
        return max(dp[size-1],dp[max2]);
    }
};

这是官方题解

class Solution {
public:
    int rob(vector<int>& nums) {
        if (nums.empty()) {
            return 0;
        }
        int size = nums.size();
        if (size == 1) {
            return nums[0];
        }
        vector<int> dp = vector<int>(size, 0);
        dp[0] = nums[0];
        dp[1] = max(nums[0], nums[1]);
        for (int i = 2; i < size; i++) {
            dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
        }
        return dp[size - 1];
    }
};

不需要dp数组的解法

class Solution {
public:
    int rob(vector<int>& nums) {
        if (nums.empty()) {
            return 0;
        }
        int size = nums.size();
        if (size == 1) {
            return nums[0];
        }
        int first = nums[0], second = max(nums[0], nums[1]);
        for (int i = 2; i < size; i++) {
            int temp = second;
            second = max(first + nums[i], second);
            first = temp;
        }
        return second;
    }
};