有一些工作: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]
思路:这道题的简单地说就是给定工人workers能接受的difficulty,然后在所有能接受的难度中选择收益最大的值,要注意的是并不是难度低的收益低,有可能难度低的收益高,所以不能对difficulty和profits排序,然后用贪心法求解,所以这道题的difficulty和profits必须要对应,我们把它们放到一个pair里,然后对pair进行排序,然后用一个变量last表示小于worker[i]中所有difficulty对应的profits的最大值(即从历史中选择worker[i]能完成的任务中对应最大的收益,用last记录,last初始化为0)。这样由于对pair和workers排序,时间复杂度(O(nlog(n)+mlog(m)),n表示pair的大小,m表示workers的大小。参考代码如下:
class Solution {
public:
int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) {
vector<pair<int, int>> memo;
for (int i = 0; i < difficulty.size(); i++) memo.push_back(make_pair(difficulty[i], profit[i]));
sort(memo.begin(), memo.end());
sort(worker.begin(), worker.end());
int last = 0;
int res = 0,j=0;
for (int i = 0; i <worker.size(); i++) {
while (j < difficulty.size() && worker[i] >= memo[j].first) {
last = max(last, memo[j].second);
j++;
}
if (j <=difficulty.size()) res += last;
}
return res;
}
};