20210319 按摩师

题目 按摩师

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

示例 1：
输入： [1,2,3,1]
输出： 4
解释： 选择 1 号预约和 3 号预约，总时长 = 1 + 3 = 4。

示例 2：
输入： [2,7,9,3,1]
输出： 12
解释： 选择 1 号预约、 3 号预约和 5 号预约，总时长 = 2 + 9 + 1 = 12。

示例 3：
输入： [2,1,4,5,3,1,1,3]
输出： 12
解释： 选择 1 号预约、 3 号预约、 5 号预约和 8 号预约，总时长 = 2 + 4 + 3 + 3 = 12。

解1

按摩师不能连续预约，所以当按摩师预约第i个顾客后，下一个预约的为第i+2个顾客或者第i+3个顾客，且起始顾客必须为第1个或第2个，可获取最大预约时间。因此通过递归实现。
另外，通过设置一个数组标记，存储以每一个顾客为起始时的最长预约时间。递归过程中，如果以i号顾客为起始的最长预约时间已经存储，则直接返回。

class Solution {
public:
    int massage(vector<int>& nums) {
    	//设置一个数组，存每个顾客起始的最长预约时间
        vector<int> flag(nums.size(),INT_MIN);
        //从第0个或第1个开始
        return max(getMax(nums,0,flag) , getMax(nums,1,flag));
    }
    int getMax(vector<int> nums , int index, vector<int> &flag)
    {
    	//如果超出数组范围，返回0
        if(index >= nums.size()) return 0;
        //数组已经存储index号顾客开始的最长预约时间
        if(flag[index] != INT_MIN)
            return flag[index];
        //看index+2或index+3号顾客起始的最长预约时间，并返回较大值
        flag[index] = max(getMax(nums,index+2,flag) , getMax(nums , index+3,flag))+nums[index];
        return flag[index];
    }
};

解2

设置两个变量表示顾客 i 的两种被选择状态下的最长预约时间：
dp0：第 i 个顾客未选中，前 i 个顾客的最长预约时间
dp1：第 i 个顾客选中，前 i 个顾客的最长预约时间

遍历数组，判断每个顾客的d0和d1，详细见注释：

class Solution {
public:
    int massage(vector<int>& nums) {
        if(nums.size() == 0) return 0;
        int i=0;
        //nums[0]受否选择中的状态
        int d0 = 0;        //nums[0]未选中
        int d1 = nums[0];  //nums[0]选中
        int t0,t1;         //临时变量
        //从nums[1]开始遍历
        for(i=1 ; i<nums.size() ; i++)
        {
            t0 = max(d0,d1); //前i个顾客的最长预约时间
            t1 = d0+nums[i]; //若选该元素，则上一个元素不应该选中，加上nums[i-1]不选的最长预约时间
            d0 = t0;
            d1 = t1;
        }
        //最后一个元素处理结束后，返回较大值
        return max(d0,d1);
    }
};