文章目录
- 前言
- 思路
- 主要思路
- 关于f函数的剖析
- Code
- 就到这,铁子们下期见!!!!
前言
铁子们好啊!今天阿辉又给大家来更新新一道好题,下面链接是23年9月27的华为笔试原题,LeetCode上面的hard
难题,阿辉带大伙来拿下它!!!
你可以安排的最多任务数目
思路
二分和单调队列以及一丢丢贪心
主要思路
先按照任务难度和工人能力排序
二分的范围是
[l,r)
左闭右开,l = 0
,r = n+1
,最多完成n
个任务,n
取任务数与工人数的较小值,因为左闭右开,所以r
取n+1
,最少完成0
个任务,所以l
取0
然后就是如何判断
l
与r
的中点m
是否是能够完成的任务数- 排序的重要就在这里体现了,我们取任务难度最小的
m
个与能力最强的m
个工人如果能够完成,那就能完成,如果不能就完成不了
- 排序的重要就在这里体现了,我们取任务难度最小的
主逻辑代码:
// 主函数,用于找出可以分配的最大任务数量。int maxTaskAssign(vector<int>& tasks, vector<int>& workers, int pills, int strength) {int n = min(tasks.size(), workers.size()); // 取任务数和工人数中较小的一个,因为任务数不能超过工人数。int l = 0; // 二分查找的左边界int r = n+1; // 二分查找的右边界sort(tasks.begin(), tasks.end()); // 将任务按难度排序sort(workers.begin(), workers.end()); // 将工人按能力排序// 二分查找,确定最大可分配任务数while (l < r) {int m = l + (r - l) / 2; // 中间点//f函数用于判断是否可以完成m个任务if (f(tasks, workers, pills, strength, m)) {l = m + 1; // 如果能完成m个任务,则尝试增加任务数} else {r = m; // 如果不能完成m个任务,则减少任务数}}return l - 1; // 返回最终的任务数(因为在二分查找结束时,l指向的是第一个不能完成的任务数)}
关于f函数的剖析
f
函数的空间复杂度是 O ( M )O(M)O(M),因为i
和j
都只前进不回退,也就是只遍历2
次m
长度的数组
N
为任务数组与工人数组的较大值
然后主函数排序两个数组是 O ( N l o g N )O(NlogN)O(NlogN),二分加上f函数最多也不超过 O ( N l o g N )O(NlogN)O(NlogN)
所以时间复杂度 O ( N l o g N )O(NlogN)O(NlogN)
空间复杂度低于 O ( N )O(N)O(N),队列长度取决于完成任务的数量
int deque[50001]; // 一个双端队列,用于存储可能通过使用或不使用药丸完成的任务。// 辅助函数,用于判断是否能在当前条件下完成m个任务。bool f(vector<int>& ts, vector<int>& ws, int p, int s, int m) {int h = 0, t = 0; // 双端队列的头部和尾部指针//i指向最容易完成的第一个任务,j指向能力第m强的工人//遍历m个最容易完成的任务以及能力最强的m个工人for (int i = 0, j = ws.size() - m; j < ws.size(); ++j) {// 遍历每一个工人,并尝试分配任务while (i < m && ts[i] <= ws[j]) {// 如果当前任务可以由工人直接完成,则将其加入队列deque[t++] = ts[i++];}//经过上面的if如果队列里面没东西,说明该试试药了//如果队列里面有东西,可能是前一个工人嗑药留下的if (h == t || ws[j] < deque[h]) {// 如果队列为空,或当前工人无法完成队列头部的任务,则尝试使用药丸--p; // 使用一颗药丸while (i < m && ts[i] <= ws[j] + s) {// 将可以通过使用药丸完成的任务加入队列deque[t++] = ts[i++];}if (h == t || p < 0 || ws[j] + s < deque[h]) {// 如果队列依然为空,或药丸用完,或即使使用药丸也无法完成队列头部的任务,则返回falsereturn false;}--t; // 上面没返回说明嗑药有用,完成最难的任务,一点子贪心各位肯定能懂,队列尾部指针前移} else {//否则++h; // 工人直接完成了队列头部的任务,队列头部指针后移}}//能走到这就说明能完成return true;
Code
class Solution {public: // 主函数,用于找出可以分配的最大任务数量。int maxTaskAssign(vector<int>& tasks, vector<int>& workers, int pills, int strength) {int n = min(tasks.size(), workers.size()); // 取任务数和工人数中较小的一个,因为任务数不能超过工人数。int l = 0; // 二分查找的左边界int r = n+1; // 二分查找的右边界sort(tasks.begin(), tasks.end()); // 将任务按难度排序sort(workers.begin(), workers.end()); // 将工人按能力排序// 二分查找,确定最大可分配任务数while (l < r) {int m = l + (r - l) / 2; // 中间点if (f(tasks, workers, pills, strength, m)) {l = m + 1; // 如果能完成m个任务,则尝试增加任务数} else {r = m; // 如果不能完成m个任务,则减少任务数}}return l - 1; // 返回最终的任务数(因为在二分查找结束时,l指向的是第一个不能完成的任务数)}int deque[50001]; // 一个双端队列,用于存储可能通过使用或不使用药丸完成的任务。// 辅助函数,用于判断是否能在当前条件下完成m个任务。bool f(vector<int>& ts, vector<int>& ws, int p, int s, int m) {int h = 0, t = 0; // 双端队列的头部和尾部指针for (int i = 0, j = ws.size() - m; j < ws.size(); ++j) {// 遍历每一个工人,并尝试分配任务while (i < m && ts[i] <= ws[j]) {// 如果当前任务可以由工人直接完成,则将其加入队列deque[t++] = ts[i++];}if (h == t || ws[j] < deque[h]) {// 如果队列为空,或当前工人无法完成队列头部的任务,则尝试使用药丸--p; // 使用一颗药丸while (i < m && ts[i] <= ws[j] + s) {// 将可以通过使用药丸完成的任务加入队列deque[t++] = ts[i++];}if (h == t || p < 0 || ws[j] + s < deque[h]) {// 如果队列依然为空,或药丸用完,或即使使用药丸也无法完成队列头部的任务,则返回falsereturn false;}--t; // 完成一个任务,队列尾部指针前移} else {++h; // 工人直接完成了队列头部的任务,队列头部指针后移}}return true; // 如果所有工人都成功分配了任务,则返回true}};