一个厨师收集了他 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
提示:
n == satisfaction.length
1 <= n <= 500
-10^3 <= satisfaction[i] <= 10^3
解题思路
显然想要最后的结果最大,那么大的数要放在右边,而小的数放在左边,所以可以先对satisfaction
从小到大排序。然后思考这样的问题,什么条件下, 删除区间[l,r]
中一个数后,最后的结果会变大?
-1 -8 0 5 -9
↑ ↑
l r
如果删除的是0
,那么5, -9
会往左移动一位,对于最后的结果就是减少了5 + (-9) + 0 * 2
,可以抽象为pre[-1] - pre[i] + i * s[i]
(其中pre
表示前缀和)。另外对于排好序的satisfactioin
,最后的结果区间一定是连续的,因为如果中间一个数被删除后得到最优解,而被删除数的左边一定比它小,那么删除它一定可以得到更大值。例如
-9 -8 -1 0 5
↑ ↑
l r
如果删除-1
可以得到最大值,那么何不删除-8
呢?所有最后的结果就是排好序的satisfactioin
的后缀数组。那么什么时候可以取到最大呢?最后的结果就是n*s[-1] + (n-1)*s[-2] +... + 1*s[i]
,我们需要知道i
要满足什么条件。
实际上可以将n*s[-1] + (n-1)*s[-2] +... + 1*s[i]
分解为(s[-1]) + (s[-1] + s[-2]) +...+ (s[-1] + ... + s[i])
。不难发现最后的结果是多组后缀和的和,显然只要s[-1] + ... + s[i] > 0
,我们可以一直向后累加。
class Solution:
def maxSatisfaction(self, s: List[int]) -> int:
res = total = 0
s.sort()
while s and s[-1] + total > 0:
total += s.pop()
res += total
return res
reference:
https://leetcode.com/problems/reducing-dishes/discuss/563384/JavaC%2B%2BPython-Easy-and-Concise
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!