Leetcode 826:安排工作以达到最大收益(超详细的解法!!!)

有一些工作:difficulty[i]表示第i个工作的难度,profit[i]表示第i个工作的收益。

现在我们有一些工人。worker[i]是第i个工人的能力,即该工人只能完成难度小于等于worker[i]的工作。

每一个工人都最多只能安排一个工作,但是一个工作可以完成多次。

举个例子,如果3个工人都尝试完成一份报酬为1的同样工作,那么总收益为3。如果一个工人不能完成任何工作,他的收益为0 。

我们能得到的最大收益是多少?

示例:

输入: difficulty = [2,4,6,8,10], profit = [10,20,30,40,50], worker = [4,5,6,7]
输出: 100 
解释: 工人被分配的工作难度是 [4,4,6,6] ,分别获得 [20,20,30,30] 的收益。

提示:

  • 1 <= difficulty.length = profit.length <= 10000
  • 1 <= worker.length <= 10000
  • difficulty[i], profit[i], worker[i] 的范围是[1, 10^5]

解题思路

这个问题的思路非常简单,采用贪心的思想。对于每个工人,找其可以满足的所有difficultyprofit最大的那个即可。具体操作如下:

首先对(difficulty, profit)进行排序,然后遍历worker,对于其中的每个工人找可以获取的最大利润。这里有一个优化,可以不用每次从头开始找。可以先对worker排序,这样我们后面的每个工人可以获得的最大利润一定是>=前一个工人的。那么此时我们只需要通过一个变量记录前一个工人的最大利润位置,每次从该位置出发即可。

class Solution:
    def maxProfitAssignment(self, difficulty: List[int], profit: List[int], worker: List[int]) -> int:
        jobs = sorted(zip(difficulty, profit))
        res = i = best = 0
        for w in sorted(worker):
            while i < len(jobs) and w >= jobs[i][0]:
                best = max(jobs[i][1], best)
                i += 1
            res += best
        return res

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

你可能感兴趣的:(leetcode解题指南,Problems)