【有序集合 堆 优先队列】1606. 找到处理最多请求的服务器|2275

本文涉及知识点

有序集合 C++堆(优先队列)
算法与数据结构汇总

LeetCode1606. 找到处理最多请求的服务器

你有 k 个服务器，编号为 0 到 k-1 ，它们可以同时处理多个请求组。每个服务器有无穷的计算能力但是 不能同时处理超过一个请求 。请求分配到服务器的规则如下：
第 i （序号从 0 开始）个请求到达。
如果所有服务器都已被占据，那么该请求被舍弃（完全不处理）。
如果第 (i % k) 个服务器空闲，那么对应服务器会处理该请求。
否则，将请求安排给下一个空闲的服务器（服务器构成一个环，必要的话可能从第 0 个服务器开始继续找下一个空闲的服务器）。比方说，如果第 i 个服务器在忙，那么会查看第 (i+1) 个服务器，第 (i+2) 个服务器等等。
给你一个 严格递增 的正整数数组 arrival ，表示第 i 个任务的到达时间，和另一个数组 load ，其中 load[i] 表示第 i 个请求的工作量（也就是服务器完成它所需要的时间）。你的任务是找到 最繁忙的服务器 。最繁忙定义为一个服务器处理的请求数是所有服务器里最多的。
请你返回包含所有 最繁忙服务器 序号的列表，你可以以任意顺序返回这个列表。
示例 1：
输入：k = 3, arrival = [1,2,3,4,5], load = [5,2,3,3,3]
输出：[1]

解释：
所有服务器一开始都是空闲的。
前 3 个请求分别由前 3 台服务器依次处理。
请求 3 进来的时候，服务器 0 被占据，所以它被安排到下一台空闲的服务器，也就是服务器 1 。
请求 4 进来的时候，由于所有服务器都被占据，该请求被舍弃。
服务器 0 和 2 分别都处理了一个请求，服务器 1 处理了两个请求。所以服务器 1 是最忙的服务器。
示例 2：
输入：k = 3, arrival = [1,2,3,4], load = [1,2,1,2]
输出：[0]
解释：
前 3 个请求分别被前 3 个服务器处理。
请求 3 进来，由于服务器 0 空闲，它被服务器 0 处理。
服务器 0 处理了两个请求，服务器 1 和 2 分别处理了一个请求。所以服务器 0 是最忙的服务器。
示例 3：
输入：k = 3, arrival = [1,2,3], load = [10,12,11]
输出：[0,1,2]
解释：每个服务器分别处理了一个请求，所以它们都是最忙的服务器。
示例 4：
输入：k = 3, arrival = [1,2,3,4,8,9,10], load = [5,2,10,3,1,2,2]
输出：[1]
示例 5：
输入：k = 1, arrival = [1], load = [1]
输出：[0]
提示：
1 <= k <= 10⁵
1 <= arrival.length, load.length <= 10⁵
arrival.length == load.length
1 <= arrival[i], load[i] <= 10⁹
arrival 保证 严格递增 。

小根堆 + 有序映射

有序集合sEmpty记录，空闲的服务器。
小根堆sEnd记录，工作中的服务器的结束时间和服务器索引，结束时可以马上完成任务。
ret[i]记录i号服务器使用次数。
枚举每个任务，其到达时间为t,用时为use，则：
while( sEnd.top() <= t ){
出栈
对应服务器加到sEmpty
}
如果sEmpty为空，忽略此任务。
auto it = sEmpty.lower(t%k);
如果it非法， it = sEmpty.begin()
ret[*it]++;
sEmpty.erase(it);
sEnd.insert(t+use);

时间复杂度： O(nlogk)

代码

class Solution {
public:
	vector<int> busiestServers(int k, vector<int>& arrival, vector<int>& load) {
		set<int> sEmpty;
		for (int i = 0; i < k; i++) {
			sEmpty.emplace(i);
		}
		vector<int> cnt(k);
		priority_queue<pair<int, int>, vector<pair<int, int>>, greater<>> end;
		for (int i = 0; i < arrival.size(); i++) {
			while (end.size() && (end.top().first <= arrival[i])) {
				sEmpty.emplace(end.top().second);
				end.pop();
			}
			if (sEmpty.empty()) { continue; }
			auto it = sEmpty.lower_bound(i % k);
			if (sEmpty.end() == it) {
				it = sEmpty.begin();
			}
			cnt[*it]++;
			end.emplace(pair<int, int>(arrival[i] + load[i], *it));
			sEmpty.erase(it);
		}
		const int iMax = *std::max_element(cnt.begin(), cnt.end());
		vector<int> res;
		for (int i = 0; i < cnt.size(); i++) {
			if (iMax == cnt[i]) {
				res.emplace_back(i);
			}
		}
		return res;
	}
};

单元测试

template<class T1, class T2>
void AssertEx(const T1& t1, const T2& t2)
{
	Assert::AreEqual(t1, t2);
}

template<class T>
void AssertEx(const vector<T>& v1, const vector<T>& v2)
{
	Assert::AreEqual(v1.size(), v2.size());
	for (int i = 0; i < v1.size(); i++)
	{
		Assert::AreEqual(v1[i], v2[i]);
	}
}

template<class T>
void AssertV2(vector<vector<T>> vv1, vector<vector<T>> vv2)
{
	sort(vv1.begin(), vv1.end());
	sort(vv2.begin(), vv2.end());
	Assert::AreEqual(vv1.size(), vv2.size());
	for (int i = 0; i < vv1.size(); i++)
	{
		AssertEx(vv1[i], vv2[i]);
	}
}

namespace UnitTest
{
	int k;
	vector<int> arrival,load;
	TEST_CLASS(UnitTest)
	{
	public:
		TEST_METHOD(TestMethod00)
		{
			k = 3, arrival = { 1,2,3,4,5 }, load = { 5,2,3,3,3 };
			auto res = Solution().busiestServers(k, arrival, load);
			AssertEx({ 1 }, res);
		}
		TEST_METHOD(TestMethod01)
		{
			k = 3, arrival = { 1,2,3,4 }, load = { 1,2,1,2 };
			auto res = Solution().busiestServers(k, arrival, load);
			AssertEx({ 0 }, res);
		}
		TEST_METHOD(TestMethod02)
		{
			k = 3, arrival = { 1,2,3 }, load = { 10,12,11 };
			auto res = Solution().busiestServers(k, arrival, load);
			AssertEx({ 0 ,1,2}, res);
		}
		TEST_METHOD(TestMethod03)
		{
			k = 3, arrival = { 1,2,3,4,8,9,10 }, load = { 5,2,10,3,1,2,2 };
			auto res = Solution().busiestServers(k, arrival, load);
			AssertEx({1 }, res);
		}
		TEST_METHOD(TestMethod04)
		{
			k = 1, arrival = { 1 }, load = { 1 };
			auto res = Solution().busiestServers(k, arrival, load);
			AssertEx({ 0 }, res);
		}	
	};
}