一个厨师收集了他 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reducing-dishes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
首先观察一下,题目给定的 n 在 500 以内,比较小所以可以直接尝试暴力法。
由题意可知,满意程度越大的菜应该放在越后面做,这样它乘上的时间就会更大。
因此不妨先把 satisfaction 从小到大排序,保障先做满意度小的菜。
然后利用双重循环暴力处理所有的组合,找到答案。
时间复杂度:O(N^2)
空间复杂度:O(1)
class Solution(object):
def maxSatisfaction(self, satisfaction):
"""
:type satisfaction: List[int]
:rtype: int
"""
res = 0
s = sorted(satisfaction)
for i in range(len(s)):
time = 1 # 从开始做菜过去的时间
tmp = 0 # 这一种做菜安排的答案
for j in range(i, len(s)):
tmp += time * s[j]
time += 1
res = max(tmp, res)
return res