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]

来源:力扣(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& difficulty, vector& profit, vector& worker) {
        int mmin = difficulty[0], ans = 0;
        vector 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]所可以获取到的最大利润

你可能感兴趣的:(LeetCode)