题目描述
一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。
思路
- 逆推法,从结尾往前推,定义一个数组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]谁大就选谁 - 正推法:动态规划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]);
}
};