有一些工作: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]
使用Hash表保存难度和难度以下的最大利润,以及二分查找lower_bound函数,注意处理边界。这道题目时间主要用在排序上。
typedef pair<int,int> PII;
class Solution {
public:
int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) {
vector<PII> temp;
int res=0;
for(int i=0;i<profit.size();i++)
temp.push_back({difficulty[i],profit[i]});
sort(temp.begin(),temp.end());
sort(difficulty.begin(),difficulty.end());
int maxprofit = 0;
unordered_map<int,int> Hash; // key 为难度 value 为当前难度以下的最大深入
for(auto it=temp.begin();it!=temp.end();++it){
if(it->second>maxprofit) maxprofit = it->second;
Hash[it->first] = maxprofit;
}
for(int i=0;i<worker.size();i++){
auto it = lower_bound(difficulty.begin(),difficulty.end(),worker[i]);
if(it==difficulty.end())
res+=Hash[*(--it)];
else if(*it==worker[i])
res+=Hash[*it];
else if(it!=difficulty.begin())
res+=Hash[*(--it)];
}
return res;
}
};