2071. Maximum Number of Tasks You Can Assign.cpp

2071. Maximum Number of Tasks You Can Assign.cpp

@@ -0,0 +1,49 @@
+class Solution {
+public:
+    int maxTaskAssign(vector<int>& tasks, vector<int>& workers, int pills, int strength) {
+        sort(tasks.begin(), tasks.end());
+        sort(workers.begin(), workers.end());
+
+        int low = 0, high = min(tasks.size(), workers.size());
+
+        while (low < high) {
+            int mid = (low + high + 1) / 2;
+            if (canAssign(tasks, workers, pills, strength, mid)) {
+                low = mid;
+            } else {
+                high = mid - 1;
+            }
+        }
+
+        return low;
+    }
+
+private:
+    bool canAssign(const vector<int>& tasks, const vector<int>& workers,
+                   int pills, int strength, int taskCount) {
+        deque<int> boosted;
+        int w = workers.size() - 1;
+        int freePills = pills;
+
+        for (int t = taskCount - 1; t >= 0; --t) {
+            int task = tasks[t];
+
+            if (!boosted.empty() && boosted.front() >= task) {
+                boosted.pop_front();
+            } else if (w >= 0 && workers[w] >= task) {
+                --w;
+            } else {
+                while (w >= 0 && workers[w] + strength >= task) {
+                    boosted.push_back(workers[w--]);
+                }
+                if (boosted.empty() || freePills == 0) {
+                    return false;
+                }
+                boosted.pop_back();
+                --freePills;
+            }
+        }
+
+        return true;
+    }
+};