有一些工作: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]
C++
class Solution {
public:
int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker)
{
int res=0;
int m=difficulty.size();
int n=worker.size();
vector<pair<int,int>> tmp;
for(int i=0;i<m;i++)
{
tmp.push_back(make_pair(difficulty[i],profit[i]));
}
sort(tmp.begin(),tmp.end());
for(int i=1;i<m;i++)
{
if(tmp[i].second<tmp[i-1].second)
{
tmp[i].second=tmp[i-1].second;
}
}
sort(worker.begin(),worker.end());
int k=0;
int val=0;
for(int i=0;i<n;i++)
{
while(k<m && worker[i]>=tmp[k].first)
{
k++;
}
if(k>=1 && k<=m)
{
res+=tmp[k-1].second;
}
}
return res;
}
};