你有 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
题目大意:计算所有工人工作的最大收益总和。
分析:设k_i为difficult数组中小于等于worker[i]的最大的difficult
(1)根据题意可知,每个工人能力确定下,其最大收益是确定的,即为max{difficult[t]},其中t为所有满足difficult[t]<=k_i的下标;
(2)由(1)可知关键在于查找worker[i]对应的k_i,因此可将每份工作按difficult数组排序,利用二分查找,确定k_i;
(3)确定k_i后需获取其最大收益,为避免重复计算相同k_i的最大收益,预先对排序后的profit数组进行一遍处理,用其存储计算每个difficult[i]的最大收益。
class Solution {
public:
int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) {
int N=profit.size(),ans=0;
vector<pair<int,int>> work;
for(int i=0;i<N;++i){
work.emplace_back(difficulty[i],profit[i]);
}
sort(work.begin(),work.end());
//预处理profit
for(int i=1;i<N;++i){
work[i].second=max(work[i].second,work[i-1].second);
}
for(int p:worker){
int l=0,r=N-1,mid;
//二分查找值<=p的最大difficult
while(l<=r){
mid=(l+r)/2;
if(work[mid].first>p) r=mid-1;
else l=mid+1;
}
if(r!=-1) ans+=work[r].second;
}
return ans;
}
};