一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。
参考
class Solution:
def massage(self, nums: List[int]) -> int:
#暴力递归,时间复杂度O(N),空间复杂度O(N)
n = len(nums)
if not n:
return(0)
if n <= 2:
return(max(nums))
dp = [nums[0]] + [max(nums[0], nums[1])] + [0]*(n-2)
for i in range(2, n):
dp[i] = max(dp[i-1], dp[i-2] + nums[i])
return(dp[-1])
上面的方法由于创建了一个dp数组,所以空间复杂度O(N),但是仔细想,例如我们在求dp[8]时,我们只需要dp[7]和dp[6]就行了,不需要创建一个数组存储之前的dp[i],所以下面的代码用pre代替dp[6],cur代替dp[7],这样就可以减小空间复杂度
参考
class Solution:
def massage(self, nums: List[int]) -> int:
#时间复杂度O(N),空间复杂度O(1)
n = len(nums)
if not n:
return(0)
if n == 1:
return(nums[0])
else:
pre = nums[0]
cur = max(nums[0], nums[1])
for i in range(2, n):
cur, pre = max(cur, pre + nums[i]), cur#同步赋值表达式首先将右侧的表达式按照顺序进行计算,然后再赋值给左侧的变量
return(cur)