题目
一个厨师收集了他 n 道菜的满意程度 satisfaction ,这个厨师做出每道菜的时间都是 1 单位时间。
一道菜的 「喜爱时间」系数定义为烹饪这道菜以及之前每道菜所花费的时间乘以这道菜的满意程度,也就是
time[i]*satisfaction[i] 。请你返回做完所有菜 「喜爱时间」总和的最大值为多少。
你可以按 任意 顺序安排做菜的顺序,你也可以选择放弃做某些菜来获得更大的总和。
示例 1:
输入:satisfaction = [-1,-8,0,5,-9] 输出:14 解释:去掉第二道和最后一道菜,最大的喜爱时间系数和为
(-11 + 02 + 5*3 = 14) 。每道菜都需要花费 1 单位时间完成。示例 2:
输入:satisfaction = [4,3,2] 输出:20 解释:按照原来顺序相反的时间做菜 (21 + 32 + 4*3 =
20)示例 3:
输入:satisfaction = [-1,-4,-5] 输出:0 解释:大家都不喜欢这些菜,所以不做任何菜可以获得最大的喜爱时间系数。
示例 4:
输入:satisfaction = [-2,5,-1,0,3,-3] 输出:35
解题思路
动态规划
以示例1(satisfaction = [-1,-8,0,5,-9])为例:
排序后:satisfaction = [-9,-8,-1,0,5]
会有如下几个值:
5×1=5
5×2+0×1=10
5×3+0×2+(-1)×1=14
5×4+0×3+(-1)×2+(-8)×1=10
5×5+0×4+(-1)×3+(-8)×2+(-9)×1=-3
最后取最大值,为14。
状态定义
将satisfaction数组进行sort以降序排列,dp[i]表示以第i个数为终点的列表的喜爱时间系数,如按降序排列后得到列表[5,0,-1,-8,-9],dp[0]表示以5为终点的列表,即dp[0]=5;
状态转移方程
使用temp加上每一次的数组satisfaction[0:i]的总和,即i=0时,temp+=satisfaction[0],i=1时,temp+=satisfaction[0]+satisfaction[1]以此类推,那么得到状态转移方程dp[i]=max(dp[i-1],dp[i-1]+temp)
代码
class Solution:
def maxSatisfaction(self, satisfaction: List[int]) -> int:
satisfaction.sort(reverse=True)
dp = [0]*len(satisfaction)
dp[0] = max(0,satisfaction[0])
temp = dp[0]
for i in range(1,len(satisfaction)):
temp += satisfaction[i]
dp[i] = max(dp[i-1],dp[i-1]+temp)
return dp[-1]