①把difficultly和profit合并为一个数组task,然后按difficultly升序,profit降序对task进行排序。
②遍历task,每个任务所能获得的最大利润为所有小于等于这个任务难度的任务的最大利润,用前缀和的思想来处理即可。
③对于每个工人,二分查找task中难度不大于这个工人能力的最大难度的任务,所对应的利润即为这个工人能获得的最大利润。
class Solution {
public:
int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) {
int n=profit.size();
vector<pair<int,int>> task(n);
for(int i=0;i<n;i++) task[i]={difficulty[i],profit[i]};
sort(task.begin(),task.end(),[](const auto &a,const auto &b){
if(a.first==b.first) return a.second>b.second;
else return a.first<b.first;
});
for(int i=1;i<n;i++){
task[i].second=max(task[i].second,task[i-1].second);
}
int ans=0;
for(auto c:worker){
if(c<task[0].first) continue;
auto p=lower_bound(task.begin(),task.end(),make_pair(c,0));
if(p==task.end()||p->first>c) p--;
ans+=p->second;
}
return ans;
}
};
时间复杂度:O(nlogn+wlogn),n为任务数量,w为工人数量
空间复杂度:O(n)