发布于2020-03-18 10:43 阅读(924) 评论(0) 点赞(13) 收藏(0)
公司有编号为 1
到 n
的 n
个工程师,给你两个数组 speed
和 efficiency
,其中 speed[i]
和 efficiency[i]
分别代表第 i
位工程师的速度和效率。请你返回由最多 k
个工程师组成的 最大团队表现值 ,由于答案可能很大,请你返回结果对 10^9 + 7
取余后的结果。
团队表现值 的定义为:一个团队中「所有工程师速度的和」乘以他们「效率值中的最小值」。
示例 1:
输入:n = 6, speed = [2,10,3,1,5,8], efficiency = [5,4,3,9,7,2], k = 2
输出:60
解释:
我们选择工程师 2(speed=10 且 efficiency=4)和工程师 5(speed=5 且 efficiency=7)。他们的团队表现值为 performance = (10 + 5) * min(4, 7) = 60 。
示例 2:
输入:n = 6, speed = [2,10,3,1,5,8], efficiency = [5,4,3,9,7,2], k = 3
输出:68
解释:
此示例与第一个示例相同,除了 k = 3 。我们可以选择工程师 1 ,工程师 2 和工程师 5 得到最大的团队表现值。表现值为 performance = (2 + 10 + 5) * min(5, 4, 7) = 68 。
示例 3:
输入:n = 6, speed = [2,10,3,1,5,8], efficiency = [5,4,3,9,7,2], k = 4
输出:72
提示:
1 <= n <= 10^5
speed.length == n
efficiency.length == n
1 <= speed[i] <= 10^5
1 <= efficiency[i] <= 10^8
1 <= k <= n
解题思路
这个问题和Leetcode 857:雇佣 K 名工人的最低成本处理思路类似,此时希望efficiency
要大,所以我们可以对所有的工程师按照efficiency
从大到小排序。然后从前向后遍历排好序的工程师,假设此时遍历到第i
个工程师(这个工程师一定是前i
个中efficiency
最小的),所以我们需要从前i-1
个工程师里面选择k-1
个speed
最大的工程师(可以通过小根堆维护)。
另外,还有一个问题需要考虑,因为题目条件是最多k
个工程师,所以上面的思路解决了恰好k
个工程师的问题,而小于k
个工程师的情况也需要考虑(也就是堆中元素个数小于k-1
的时候也要计算最大团队表现值)。
class Solution:
def maxPerformance(self, n: int, speed: List[int], efficiency: List[int], k: int) -> int:
heap = []
res = sSum = 0
for e, s in sorted(zip(efficiency, speed), reverse=True):
sSum += s
res = max(res, sSum * e)
heapq.heappush(heap, s)
if len(heap) >= k: sSum -= heapq.heappop(heap)
return res % (10**9 + 7)
这个问题的思维难度比Leetcode 857:雇佣 K 名工人的最低成本简单。
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!
作者:战天
链接:https://www.pythonheidong.com/blog/article/265151/ac4411792fcebdfc4aef/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!