Leetcode之 按摩师

题目描述

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

思路

1. 逆推法，从结尾往前推，定义一个数组sum_max，其中sum_max[i]表示从i位置之后的最长总预约时长，我们发现:
   sum_max[i] = nums[i] + max(sum_max[i+2],…sum_max[len-1])，则最后递推至起点，判断sum_max[0]和sum_max[1]谁大就选谁
2. 正推法：动态规划dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])

代码

方法一：

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