程序员面试金典 面试题 17.16. 按摩师

LeetCode刷题笔记 小偷三题 打家劫舍系列
和小偷系列第一题完全相同。

1.直男解法

两种状态
【0】表示当天不接受预约
【1】表示当天接受预约

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

100%, 空间从O(n)优化到O(1)，包会～

O(n)优化

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

O(1)优化

class Solution {
public:
    int massage(vector<int>& nums) {
        int a=0,b=0;
        for(int i=0;i<nums.size();i++){
            int c=max(b,a+nums[i]);
            a=b;
            b=c;
        }
        return b;
    }
};