Leetcode 826. 安排工作以达到最大收益

有一些工作：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;
        
    }
};