面试题 17.16. 按摩师

题目

一开始以为是用递归，想了半天没想出来，偷看了一下答案。答案的思路跟递归类似，假设在当前 时刻， 为当前预约不接的情况下最长预约时间， 则为接受当前预约的最长预约时间。

那很显然，由于不能接受相邻两个预约，

不接受当前预约的话，上一个预约接不接受都可以，

最后只要比较两种情况即可 

1
2
3
4
5
6
7
8
9
10
11
class Solution:
  def massage(self, nums) -> int:
    if not nums:
      return 0
    n = len(nums)
    not_choose = 0
    choose = 0
    for n in nums:
      not_choose, choose = max(not_choose, choose), not_choose+n
    return max(not_choose, choose)
 # 52 ms	13.6 MB

这种问题原来有个名字叫动态规划，上面推导的方程叫状态转移方程，可以找找资料来看一下。