第九届蓝桥杯 Java B—— 日志统计

小明维护着一个程序员论坛。现在他收集了一份 "点赞" 日志，日志共有 N 行。其中每一行的格式是：

ts id

表示在 ts 时刻编号 id 的帖子收到一个 "赞"。

现在小明想统计有哪些帖子曾经是 "热帖"。如果一个帖子曾在任意一个长度为 D 的时间段内收到不少于 K 个赞，小明就认为这个帖子曾是 "热帖"。

具体来说，如果存在某个时刻 T 满足该帖在 [T, T+D) 这段时间内 (注意是左闭右开区间) 收到不少于 K 个赞，该帖就曾是 "热帖"。

给定日志，请你帮助小明统计出所有曾是 "热帖" 的帖子编号。

【输入格式】
第一行包含三个整数 N、D 和 K。
以下 N 行每行一条日志，包含两个整数 ts 和 id。

对于 50% 的数据，1 <= K <= N <= 1000
对于 100% 的数据，1 <= K <= N <= 100000 0 <= ts <= 100000 0 <= id <= 100000

【输出格式】
按从小到大的顺序输出热帖 id。每个 id 一行。

【输入样例】
7 10 2
0 1
0 10
10 10
10 1
9 1
100 3
100 3

【输出样例】
1
3

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU 消耗 < 1000ms

import java.io.BufferedInputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeMap;

public class Main {

    public static void main(String[] args) {
        Scanner cin = new Scanner(new BufferedInputStream(System.in));
        int n = cin.nextInt();
        int d = cin.nextInt();
        int k = cin.nextInt();
        Map<Integer, ArrayList<Integer>> map = new TreeMap<Integer, ArrayList<Integer>>();
        for (int i = 0; i < n; i++) {
            int ts = cin.nextInt();
            int id = cin.nextInt();
            ArrayList<Integer> list = new ArrayList<Integer>();
            if (!map.containsKey(id)) {
                list.add(ts);
                map.put(id, list);
            } else {
                list = map.get(id);
                list.add(ts);
                map.put(id, list);
            }
        }
        

        Set<Integer> res = new HashSet<Integer>();
        for (int id : map.keySet()) {
            Collections.sort(map.get(id));
            int sum = 1;
            int start = map.get(id).get(0);
            for (int i = 1; i < map.get(id).size(); i++) {
                if (map.get(id).get(i) - start < d) {
                    sum++;
                } else if (map.get(id).get(i) - start > d) {
                    start = map.get(id).get(i);
                } if (sum >= k) {
                    res.add(id);
                    break;
                }
            }
        }
        for (int i : res)
            System.out.println(i);
    }
}