给你一个整数 n
,表示服务器的总数目,再给你一个下标从 0 开始的 二维 整数数组 logs
,其中 logs[i] = [server_id, time]
表示 id 为 server_id
的服务器在 time
时收到了一个请求。
同时给你一个整数 x
和一个下标从 0 开始的整数数组 queries
。
请你返回一个长度等于 queries.length
的数组 arr
,其中 arr[i]
表示在时间区间 [queries[i] - x, queries[i]]
内没有收到请求的服务器数目。
注意时间区间是个闭区间。
示例 1:
**输入:** n = 3, logs = [[1,3],[2,6],[1,5]], x = 5, queries = [10,11]
**输出:** [1,2]
**解释:**
对于 queries[0]:id 为 1 和 2 的服务器在区间 [5, 10] 内收到了请求,所以只有服务器 3 没有收到请求。
对于 queries[1]:id 为 2 的服务器在区间 [6,11] 内收到了请求,所以 id 为 1 和 3 的服务器在这个时间段内没有收到请求。
示例 2:
**输入:** n = 3, logs = [[2,4],[2,1],[1,2],[3,1]], x = 2, queries = [3,4]
**输出:** [0,1]
**解释:**
对于 queries[0]:区间 [1, 3] 内所有服务器都收到了请求。
对于 queries[1]:只有 id 为 3 的服务器在区间 [2,4] 内没有收到请求。
提示:
1 <= n <= 105
1 <= logs.length <= 105
1 <= queries.length <= 105
logs[i].length == 2
1 <= logs[i][0] <= n
1 <= logs[i][1] <= 106
1 <= x <= 105
x < queries[i] <= 106
为方便回答询问,可以把 logs 的时间和询问都从小到大排序。对于询问,为了不打乱顺序,可以创建一个下标数组对其排序。
由于询问的窗口大小是固定的,所以可以用滑动窗口(双指针)来算,维护窗口内的各个服务器收到了多少次请求 cnt,以及没有收到请求的服务器数目 outOfRange。
具体见【双周赛 107】 第四题讲解,欢迎点赞投币!
[sol-Python3]1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
| class Solution: def countServers(self, n: int, logs: List[List[int]], x: int, queries: List[int]) -> List[int]: logs.sort(key=lambda p: p[1]) ans = [0] * len(queries) cnt = [0] * (n + 1) out_of_range = n left = right = 0 for qi, q in sorted(enumerate(queries), key=lambda p: p[1]): while right < len(logs) and logs[right][1] <= q: i = logs[right][0] if cnt[i] == 0: out_of_range -= 1 cnt[i] += 1 right += 1 while left < len(logs) and logs[left][1] < q - x: i = logs[left][0] cnt[i] -= 1 if cnt[i] == 0: out_of_range += 1 left += 1 ans[qi] = out_of_range return ans
|
[sol-Java]1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
| class Solution { public int[] countServers(int n, int[][] logs, int x, int[] queries) { int nq = queries.length; var id = new Integer[nq]; for (int i = 0; i < nq; i++) id[i] = i; Arrays.sort(id, (i, j) -> queries[i] - queries[j]); Arrays.sort(logs, (a, b) -> a[1] - b[1]);
int[] ans = new int[nq], cnt = new int[n + 1]; int outOfRange = n, left = 0, right = 0; for (int i : id) { while (right < logs.length && logs[right][1] <= queries[i]) if (cnt[logs[right++][0]]++ == 0) outOfRange--; while (left < logs.length && logs[left][1] < queries[i] - x) if (--cnt[logs[left++][0]] == 0) outOfRange++; ans[i] = outOfRange; } return ans; } }
|
[sol-C++]1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
| class Solution { public: vector<int> countServers(int n, vector<vector<int>> &logs, int x, vector<int> &queries) { int nq = queries.size(), id[nq], cnt[n + 1]; memset(cnt, 0, sizeof(cnt)); iota(id, id + nq, 0); sort(id, id + nq, [&](int i, int j) { return queries[i] < queries[j]; }); sort(logs.begin(), logs.end(), [](const auto &a, const auto &b) { return a[1] < b[1]; });
vector<int> ans(nq); int out_of_range = n, left = 0, right = 0; for (int i: id) { while (right < logs.size() && logs[right][1] <= queries[i]) if (cnt[logs[right++][0]]++ == 0) out_of_range--; while (left < logs.size() && logs[left][1] < queries[i] - x) if (--cnt[logs[left++][0]] == 0) out_of_range++; ans[i] = out_of_range; } return ans; } };
|
[sol-Go]1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
| func countServers(n int, logs [][]int, x int, queries []int) []int { type pair struct{ q, i int } qs := make([]pair, len(queries)) for i, q := range queries { qs[i] = pair{q, i} } sort.Slice(qs, func(i, j int) bool { return qs[i].q < qs[j].q }) sort.Slice(logs, func(i, j int) bool { return logs[i][1] < logs[j][1] })
ans := make([]int, len(queries)) cnt := make([]int, n+1) outOfRange, left, right := n, 0, 0 for _, p := range qs { for ; right < len(logs) && logs[right][1] <= p.q; right++ { i := logs[right][0] if cnt[i] == 0 { outOfRange-- } cnt[i]++ } for ; left < len(logs) && logs[left][1] < p.q-x; left++ { i := logs[left][0] cnt[i]-- if cnt[i] == 0 { outOfRange++ } } ans[p.i] = outOfRange } return ans }
|
复杂度分析
- 时间复杂度:\mathcal{O}(n + m\log m+ q\log q),其中 m 为 logs 的长度,q 为 queries 的长度。用数组统计出现次数,复杂度要加 n。
- 空间复杂度:\mathcal{O}(n+q)。
思考题
如果询问的区间长度不固定,要怎么做?也就是输入的询问是 [\textit{left}_i, \textit{right}_i]。