面试题 17.16. 按摩师
一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。注意:本题相对原题稍作改动
示例 1:
输入: [1,2,3,1] 输出: 4 解释: 选择 1 号预约和 3 号预约,总时长 = 1 + 3 = 4。
示例 2:
输入: [2,7,9,3,1] 输出: 12 解释: 选择 1 号预约、 3 号预约和 5 号预约,总时长 = 2 + 9 + 1 = 12。
示例 3:
输入: [2,1,4,5,3,1,1,3] 输出: 12 解释: 选择 1 号预约、 3 号预约、 5 号预约和 8 号预约,总时长 = 2
- 4 + 3 + 3 = 12。
一开始还是想简单了,直接写了一个单双号(脸上写满了尴尬),后来发现可以跳号,于是就不行了,那么我们用一种动态规划的办法,对于每个时刻(除了第一时刻和第二时刻的预约我们可以直接放上去,因为这两个时刻我们要用做跳号的基准位置)的当前最大时间有三种情况。
1:前一时刻未预约作为休息时间,当前时长为:dp[i-2]+nums[i]
2:退掉前一时刻的预约,当前时刻为:dp[i-1]-nums[i-1]+nums[i]
3: 将这次作为休息时间不预约:dp[i-1]
于是,我们只要找出每个时刻最大的时长,合并在一起就是整个预约时间段最大的时长
代码如下:
class Solution:
def massage(self, nums: List[int]) -> int:
if(nums==[]):
return 0
if(len(nums)==1):
return nums[0]
dp=[0 for i in range(len(nums))]
dp[0]=nums[0]
dp[1]=nums[1]
for i in range(2,len(nums)):
dp[i]=max(dp[i-2]+nums[i],dp[i-1]-nums[i-1]+nums[i],dp[i-1])
return max(dp[-1],dp[0])#有零是因为当第二时刻往后不存在时,第一时刻会可能大于的第二时刻
动规稳得一批