题目
你有 n 个工作和 m 个工人。给定三个数组: difficulty, profit 和 worker ,其中:
difficulty[i] 表示第 i 个工作的难度,profit[i] 表示第 i 个工作的收益。
worker[i] 是第 i 个工人的能力,即该工人只能完成难度小于等于 worker[i] 的工作。
每个工人 最多 只能安排 一个 工作,但是一个工作可以 完成多次 。
举个例子,如果 3 个工人都尝试完成一份报酬为 $1 的同样工作,那么总收益为 $3 。如果一个工人不能完成任何工作,他的收益为 $0 。
返回 在把工人分配到工作岗位后,我们所能获得的最大利润 。
示例 1:
输入: 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] 的收益。
示例 2:
输入: difficulty = [85,47,57], profit = [24,66,99], worker = [40,25,25]
输出: 0
提示:
n == difficulty.length
n == profit.length
m == worker.length
1 <= n, m <= 104
1 <= difficulty[i], profit[i], worker[i] <= 105
题解
先将难度和收益进行捆绑
然后对其进行排序,排序之后对收益进行处理,让收益始终为目前难度下的最大收益
对工人进行排序
从高难度开始遍历,对于每个难度都在工人中使用二分法找到能够干活且未被分配的人
class Solution {
public:
int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) {
vector<pair<int,int>> jobs;
int n = difficulty.size();
int m = worker.size();
for(int i=0;i<n;i++)
{
jobs.emplace_back(difficulty[i],profit[i]);
}
sort(jobs.begin(),jobs.end(),[](pair<int,int> &a,pair<int,int> &b){return a.first <= b.first;});
int max_profit = 0;
for(int i=0;i<n;i++)
{
max_profit = max(max_profit,jobs[i].second);
jobs[i].second = max_profit;
}
sort(worker.begin(),worker.end());
int res = 0;
int pre = 0;//记录已经被分配任务的工人数
for(int i = n-1;i>=0;i--)
{
int cnt = worker.end() - lower_bound(worker.begin(),worker.end(),jobs[i].first);//大于等于本难度的工人人数
//cout<<jobs[i].first<<" "<<jobs[i].second<<" "<<cnt-pre<<endl;
res += (cnt-pre)*jobs[i].second;
pre = cnt;
if(pre == m)
break;
}
return res;
}
};