蓝桥杯之日志统计

标题：日志统计

小明维护着一个程序员论坛。现在他收集了一份"点赞"日志，日志共有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

思路：
首先按id号进行排序，如果id号相同就按照时间进行排序。然后用一个HashMap集合存储id号和每个id号的起始位置的下标，再用一个HashMap集合存储id号和每个id号对应的终止位置的下标，再用一个HashSet存储id号（方便之后遍历，或者也可以直接用之前的HashMap），然后用滑动窗口法去判断每个id号是否符合题目要求。

注:把索引左闭右开区间[left, right)称为一个窗口。

import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Scanner;
import java.util.TreeSet;

public class LanQiao60 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int d = sc.nextInt();
        int k = sc.nextInt();
        int[][] nums = new int[n][2];
        for(int i = 0; i < n; i++){
            nums[i][1] = sc.nextInt();//输入id号
            nums[i][0] = sc.nextInt();//输入第几秒收到的赞
        }
        Arrays.sort(nums, new Comparator<int[]>() {
            @Override
            public int compare(int[] a, int[] b) {
                if(a[0] == b[0]){
                    return a[1] - b[1];
                }
                return a[0] - b[0];
            }
        });
        /*for(int i = 0; i < n; i++){
            for(int j = 0; j < 2; j++){
                System.out.print(nums[i][j] + " ");
            }
            System.out.println();
        }*/
        HashMap<Integer, Integer> map = new HashMap<>();//记录每一种id号以及它的首元素下标
        HashMap<Integer, Integer> map1 = new HashMap<>();//记录每一种id号以及它的尾元素下标
        HashSet<Integer> set = new HashSet<>();//记录每一种id号
        for(int i = 0; i < n; i++){
            if(!map.containsKey(nums[i][0])){
                map.put(nums[i][0], i);
            }
            map1.put(nums[i][0], i);
            set.add(nums[i][0]);
        }

        TreeSet<Integer> res = new TreeSet<>();
        for(int id : set){
            if(isValid(id, map, map1, k, nums, d)){
                res.add(id);
            }
        }
        for(int num : res){
            System.out.println(num);
        }
    }
    public static boolean isValid(int id, HashMap<Integer, Integer> map, HashMap<Integer, Integer> map1, int k, int[][] nums, int d){
        int start = map.get(id);
        int end = map1.get(id);
        int right = start, left = start;
        int count = 0;
        while(right <= end){
            int num = nums[right][1];
            right++;
            if(num - nums[left][1] < d){
                count++;
                if(count >= k){
                    return true;
                }
            }
            while(num - nums[left][1] >= d){
                left++;
                if(num - nums[left][1] >= d){
                    count--;
                }
            }
        }
        return false;
    }
}