LeetCode 826. 安排工作以达到最大收益(C++)

有一些工作: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]

C++

class Solution {
public:
    int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) 
    {
        int res=0;
        int m=difficulty.size();
        int n=worker.size();
        vector<pair<int,int>> tmp;
        for(int i=0;i<m;i++)
        {
            tmp.push_back(make_pair(difficulty[i],profit[i]));
        }
        sort(tmp.begin(),tmp.end());
        for(int i=1;i<m;i++)
        {
            if(tmp[i].second<tmp[i-1].second)
            {
                tmp[i].second=tmp[i-1].second;
            }
        }
        sort(worker.begin(),worker.end());
        int k=0;
        int val=0;
        for(int i=0;i<n;i++)
        {
            while(k<m && worker[i]>=tmp[k].first)
            {
                k++;
            }
            if(k>=1 && k<=m)
            {
                res+=tmp[k-1].second;
            }
        }      
        return res;        
    }
};

猜你喜欢

转载自blog.csdn.net/qq_27060423/article/details/88074953