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

题目链接描述示例初始代码模板代码

题目链接

https://leetcode-cn.com/problems/find-servers-that-handled-most-number-of-requests/

描述

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

示例

示例 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]

初始代码模板

class Solution { public List<Integer> busiestServers(int k, int[] arrival, int[] load) { } }

代码

20201003双周赛第四题，不会用TreeSet和TreeMap，只好看了看大佬代码，代码配合注释如下 不当之处可以直接评论或者私信我

class Solution { public List<Integer> busiestServers(int k, int[] arrival, int[] load) { //长度为2的数组，保存服务器编号以及任务完成时间，根据完成时间进行升序排列 PriorityQueue<int[]> queue = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]); //保存可利用的服务器 TreeSet<Integer> set = new TreeSet<>(); //记录每个服务器处理的任务数 int[] counts = new int[k]; //开始时所有的服务器都可以用 for (int i = 0; i < k; i++) { set.add(i); } for (int i = 0; i < arrival.length; i++) { //任务完成，把相应的服务器添加进可利用的服务器集合，即set while (!queue.isEmpty() && queue.peek()[1] <= arrival[i]) { set.add(queue.poll()[0]); } //如果这个任务到达时没有可以利用的服务器，直接舍弃，有的话才可以处理 if (!set.isEmpty()) { //刚开始寻找的时候找i % k的服务器，没有则往后找 //ceiling​(E e) 返回此set中大于或等于给定元素的元素，如果没有这样的元素，则返回null 。 Integer find = set.ceiling(i % k); //后边没找到，即 i % k 到 k - 1没找到， 找0 到 i % k - 1 if (find == null) { //由于set就是升序排列的，返回最小的就行了，有了Treeset就不用挨个遍历了 int num = set.first(); set.remove(num); queue.offer(new int[]{num, arrival[i] + load[i]}); counts[num]++; } else { //找到了就用find计算 set.remove(find); queue.offer(new int[]{find, arrival[i] + load[i]}); counts[find]++; } } } return maxTast(counts); } //寻找完成数目最多的服务器 private List<Integer> maxTast(int[] counts){ //首先寻找最大值 int max = counts[0]; for (int i : counts) { max = Math.max(max, i); } List<Integer> ans = new LinkedList<>(); for (int i = 0; i < counts.length; i++) { if (counts[i] == max) { ans.add(i); } } return ans; } }