给你 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;
}
};