【C++二分查找贪心】826. 安排工作以达到最大收益-CSDN博客

本文链接：https://blog.csdn.net/he_zhidan/article/details/140902365

本文涉及的基础知识点

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

你有 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 <= 10⁴
1 <= difficulty[i], profit[i], worker[i] <= 10⁵

二分查找+贪心

则有如下性质：
任何工人，如果能完成工作，则一定完成工作。如果能完成多项工作，则完成价值最高的工作。

方法一：

枚举工人，计算能完成价值最高的工作。时间复杂度：O(mlogn)
如果工作a比份工作b：难度大，价值低，则一定存在最优解，不包括a，可以用决策包容性证明。即可以淘汰a，淘汰所有a后，价值和难度都有序。如果有模板直接使用。#

方法二：

按工作价值从高到低排序。每功份工作，都让能完成此工作能力最低的人完成。时间复杂度：O(nlogm)
multiset setWorker(worker.begin(),worker.end());
multimap<int, int,greater<>> mJob;

代码

核心代码

class Solution {
		public:
			int maxProfitAssignment(vector<int>& difficulty, vector<int>& profit, vector<int>& worker) {
				multiset<int> setWorker(worker.begin(),worker.end());
				multimap<int, int,greater<>> mJob;
				for (int i = 0; i < profit.size(); i++) {
					mJob.emplace(profit[i], difficulty[i]);
				}
				int ret = 0;
				for ( auto& [profit, diff] : mJob) {
					auto it = setWorker.lower_bound(diff);
					ret += profit * distance(it,setWorker.end());
					setWorker.erase(it, setWorker.end());
				}
				return ret;
			}			
		};

单元测试

	TEST_METHOD(TestMethod1)
		{
			difficulty = { 1,2 }, profit = { 1,1 }, worker = { 1,2 };
			auto res = Solution().maxProfitAssignment(difficulty, profit, worker);
			AssertEx(2, res);
		}
		TEST_METHOD(TestMethod2)
		{
			difficulty = { 2,1 }, profit = { 1,1 }, worker = { 1,2 };
			auto res = Solution().maxProfitAssignment(difficulty, profit, worker);
			AssertEx(2, res);
		}
		TEST_METHOD(TestMethod3)
		{
			difficulty = { 1}, profit = { 1}, worker = { 1,1 };
			auto res = Solution().maxProfitAssignment(difficulty, profit, worker);
			AssertEx(2, res);
		}
		TEST_METHOD(TestMethod11)
		{
			difficulty = { 2,4,6,8,10 }, profit = { 10,20,30,40,50 }, worker = { 4,5,6,7 };
			auto res = Solution().maxProfitAssignment(difficulty, profit, worker);
			AssertEx(100, res);
		}
		TEST_METHOD(TestMethod12)
		{
			difficulty = { 85,47,57 }, profit = { 24,66,99 }, worker = { 40,25,25 };
			auto res = Solution().maxProfitAssignment(difficulty, profit, worker);
			AssertEx(0, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功