Leetcode 面试题 17.16. 按摩师

问题描述

一个有名的按摩师会收到源源不断的预约请求，每个预约都可以选择接或不接。在每次预约服务之间都要休息时间，因此她不能接受相邻的预约。给定一个预约请求序列，替按摩师找到最优的预约集合（总预约时间最长），返回总的分钟数。
$输入：[1,2,3,1]$
$输出：4$

解题报告

• 动态规划
  当我们回看转移方程时会发现计算 $dp[i][0/1]$ 时，只与前一个状态 $dp[i-1][0/1]$ 有关，所以我们不用开数组，只用两个变量 $d{p}_0$，$d{p}_1$ 分别存储 $dp[i-1][0]$ 和 $dp[i-1][1]$ 的答案，然后去转移更新答案即可

实现代码

• 开 $O(N)$ 的空间大小

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

• 开 $O(1)$ 的空间大小

class Solution{
    public:
        int massage(vector<int>&nums){
            int n=nums.size();
            if(!n) return 0;
            int dp0=0,dp1=nums[0],temp0,temp1;
            for(int i=1;i<n;i++){
                temp0=max(dp0,dp1);
                temp1=dp0+nums[i];

                dp0=temp0;
                dp1=temp1;
            }
            return max(dp0,dp1);
        }
};