1402.做菜顺序
一个厨师收集了他n道菜的满意程度satisfaction ,这个厨师做出每道菜的时间都是1单位时间。
一道菜的 「喜爱时间」系数定义为烹饪这道菜以及之前每道菜所花费的时间乘以这道菜的满意程度,
也就是time[i] * satisfaction[i] 。
请你返回做完所有菜 「喜爱时间」总和的最大值为多少。
你可以按
任意
顺序安排做菜的顺序,你也可以选择放弃做某些菜来获得更大的总和。
示例1:
输入:satisfaction = [-1, -8, 0, 5, -9]
输出:14
解释:去掉第二道和最后一道菜,最大的喜爱时间系数和为(-1 * 1 + 0 * 2 + 5 * 3 = 14) 。每道菜都需要花费1单位时间完成。
示例2:
输入:satisfaction = [4, 3, 2]
输出:20
解释:按照原来顺序相反的时间做菜(2 * 1 + 3 * 2 + 4 * 3 = 20)
示例3:
输入:satisfaction = [-1, -4, -5]
输出:0
解释:大家都不喜欢这些菜,所以不做任何菜可以获得最大的喜爱时间系数。
示例4:
输入:satisfaction = [-2, 5, -1, 0, 3, -3]
输出:35
def func(satisfaction):
# 方法一,利用列表相乘再相加
# n = len(satisfaction)
# res = 0
# times = [i for i in range(1, n+1)]
# satisfaction.sort(reverse=False)
# for i in range(n):
# n = [a*b for a, b in zip(times, satisfaction)]
# temp = sum(list(n))
# if temp > 0 and temp > res:
# res = temp
# times, satisfaction = times[:-1], satisfaction[1:]
# return res
# 方法二,累加
# satisfaction.sort(reverse=True)
# presum, ans = 0, 0
# for si in satisfaction:
# if presum + si > 0:
# presum += si
# ans += presum
# print(presum, ans)
# else:
# break
# return ans
# 方法三,利用itertools.accumulate实现累加
from itertools import accumulate
return max(accumulate(accumulate([0] + sorted(satisfaction, reverse=True))))
print(func(satisfaction = [-2, -3]))