Leetcode198. 打家劫舍 动态规划.线性动态规划.单串问题

状态转移

设问题规模为i,设dp[i]为处理[0,i]的问题时的最大收入. 当i==n时,易得结果.
dp[i]==max(dp[i-1],dp[i-2]+nums[i])
通过是否打劫第i家把问题分成两个区域.
又因为dp当前结果只和前两次dp的结果有关,所以可以压缩内存使用长度到常量.
对于基础问题有:

 dp[0]=nums[0];
 dp[1]=max(nums[0],nums[1]);
 

代码

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