有一些工作: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]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/most-profit-assigning-work
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
typedef struct Node{
int dif, pro;
Node(int a, int b):dif(a), pro(b){}
}node;
static bool cmp(node a, node b){
if(a.dif == b.dif) return a.pro < b.pro;
return a.dif < b.dif;
}
int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) {
int mmin = difficulty[0], ans = 0;
vector<node> f;
for(int i = 1;i < difficulty.size();i++)
if(mmin > difficulty[i]) mmin = difficulty[i];
for(int i = 0;i < difficulty.size();i++){
f.push_back(Node(difficulty[i], profit[i]));
}
sort(f.begin(), f.end(), cmp);
sort(worker.begin(), worker.end());
int windex = 0, findex = 0, maxnum = f[0].pro;
while(windex < worker.size()){
while(findex != f.size() - 1 && f[findex + 1].dif <= worker[windex]){
maxnum = max(f[findex].pro, maxnum);
findex++;
}
maxnum = max(f[findex].pro, maxnum);
if(worker[windex] >= mmin)
ans += maxnum;
windex++;
}
return ans;
}
};
解题思路:利用结构体将difficulty和profit绑定在一起排序,将worker排序(顺序不影响最后结果所以不需要其他处理)。利用两个指针来扫描两个部分,maxnum是用来保存小于worker[windex]所可以获取到的最大利润