LeetCode 5913. 你可以安排的最多任务数目

给你 n 个任务和 m 个工人。每个任务需要一定的力量值才能完成,需要的力量值保存在下标从 0 开始的整数数组 tasks 中,第 i 个任务需要 tasks[i] 的力量才能完成。每个工人的力量值保存在下标从 0 开始的整数数组 workers 中,第 j 个工人的力量值为 workers[j] 。每个工人只能完成 一个 任务,且力量值需要 大于等于 该任务的力量要求值(即 workers[j] >= tasks[i] )。

除此以外,你还有 pills 个神奇药丸,可以给 一个工人的力量值 增加 strength 。你可以决定给哪些工人使用药丸,但每个工人 最多 只能使用 一片 药丸。

给你下标从 0 开始的整数数组tasks 和 workers 以及两个整数 pills 和 strength ,请你返回 最多 有多少个任务可以被完成。

示例 1:

输入:tasks = [3,2,1], workers = [0,3,3], pills = 1, strength = 1
输出:3
解释:
我们可以按照如下方案安排药丸:
- 给 0 号工人药丸。
- 0 号工人完成任务 2(0 + 1 >= 1)
- 1 号工人完成任务 1(3 >= 2)
- 2 号工人完成任务 0(3 >= 3)
示例 2:

输入:tasks = [5,4], workers = [0,0,0], pills = 1, strength = 5
输出:1
解释:
我们可以按照如下方案安排药丸:
- 给 0 号工人药丸。
- 0 号工人完成任务 0(0 + 5 >= 5)
示例 3:

输入:tasks = [10,15,30], workers = [0,10,10,10,10], pills = 3, strength = 10
输出:2
解释:
我们可以按照如下方案安排药丸:
- 给 0 号和 1 号工人药丸。
- 0 号工人完成任务 0(0 + 10 >= 10)
- 1 号工人完成任务 1(10 + 10 >= 15)
示例 4:

输入:tasks = [5,9,8,5,9], workers = [1,6,4,2,6], pills = 1, strength = 5
输出:3
解释:
我们可以按照如下方案安排药丸:
- 给 2 号工人药丸。
- 1 号工人完成任务 0(6 >= 5)
- 2 号工人完成任务 2(4 + 5 >= 8)
- 4 号工人完成任务 3(6 >= 5)
 

提示:

n == tasks.length
m == workers.length
1 <= n, m <= 5 * 104
0 <= pills <= m
0 <= tasks[i], workers[j], strength <= 109

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-number-of-tasks-you-can-assign
 

解法:

二分加贪心。像这种hard题,都是至少两种算法结合。一开始就用二分法去固定完成任务的数量这是没想到了。如果一开始就知道一定可以完成k个任务,那么最好的方式就是用力量最强的k个人去完成最简单的k个任务。k个任务的完成用的是贪心思想:

1.不使用药丸的情况下:最强的人去完成最难的任务;

2.需要使用药丸:让尽可能恰好能完成任务的人吃;

找个时间好好研究下二分。

class Solution {


public:
	int maxTaskAssign(vector<int>& tasks, vector<int>& workers, int pills, int strength) {
		int n = tasks.size(), m = workers.size();
		sort(tasks.begin(), tasks.end());
		sort(workers.begin(), workers.end());
		auto check = [&](int k)->bool{
			deque<int> workersQ;
			int p = pills;
			for (int i = m-k; i <= m-1; i++)
				workersQ.push_back(workers[i]);
			for (int i = k - 1; i >= 0; i--)
			{
				if (tasks[i] <= workersQ.back())
				{
					workersQ.pop_back();
				}
				else if (p > 0)
				{
					p--;
					int target = tasks[i] - strength;
					deque<int>::iterator iter = lower_bound(workersQ.begin(), workersQ.end(), target);
					if (iter != workersQ.end())
					{
						workersQ.erase(iter);
					}
					else
						return false;
				}
				else
					return false;
			}
			return true;
		};

		int maxC = min(n, m);
		int left = 0, right = maxC;
		int ans = 0;
		while (left < right)
		{
			int k = left + (right - left + 1) / 2;
			if (check(k))
			{
				ans = k;
				left = k;
			}
			else
				right = k - 1;
		}
		if (check(left))    //多了一次
			return left;
		else
			return left - 1;
	}
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值