每日一题,每日一练,9按摩师(这是动态规划还是贪心来着)

面试题 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])#有零是因为当第二时刻往后不存在时,第一时刻会可能大于的第二时刻

动规稳得一批

发布了10 篇原创文章 · 获赞 0 · 访问量 88

猜你喜欢

转载自blog.csdn.net/final_begin/article/details/105065611