题目
一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。
注意:本题相对原题稍作改动
示例 1:
输入: [1,2,3,1]
输出: 4
解释: 选择 1 号预约和 3 号预约,总时长 = 1 + 3 = 4。
解题思路
动态规划
- f(i)表示第i次预约时候,最优的预约集合
- f(0) = nums[0]
- f(1) = max(nums[0], nums[1])
- 动态转移方程:f(i) = max{f(i-2) + nums[i], f(i-1)}
代码
class Solution {
public int massage(int[] nums) {
if (nums == null || nums.length == 0) return 0;
if (nums.length == 1) return nums[0];
int p = nums[0], q = Math.max(nums[0], nums[1]);
for (int i = 2; i < nums.length; i++) {
int tmp = Math.max(p + nums[i], q);
p = q;
q = tmp;
}
return q;
}
}