【LeetCode】826. 安排工作以达到最大收益

题目

你有 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;
    }
};