面试题 17.16. 按摩师

题目

一个有名的按摩师会收到源源不断的预约请求，每个预约都可以选择接或不接。在每次预约服务之间要有休息时间，因此她不能接受相邻的预约。给定一个预约请求序列，替按摩师找到最优的预约集合（总预约时间最长），返回总的分钟数。
注意：本题相对原题稍作改动
示例 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;
  }
}

题目来源：力扣（LeetCode）