题目描述
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class Main {
static int N=100010;
static BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
static PrintWriter out=new PrintWriter(new OutputStreamWriter(System.out));
static boolean[] isOk=new boolean[N];
//记录区间内出现的次数
static int[] cnt=new int[N];
public static void main(String[] args) throws IOException {
String[] s = br.readLine().split(" ");
int n=Integer.parseInt(s[0]);
int d=Integer.parseInt(s[1]);
int k=Integer.parseInt(s[2]);
Map<Integer, List<Integer>> map=new HashMap<>();
for (int i=0;i<n;i++){
s=br.readLine().split(" ");
int ts=Integer.parseInt(s[0]);
int id=Integer.parseInt(s[1]);
if (!map.containsKey(ts)) map.put(ts,new ArrayList<>());
map.get(ts).add(id);
}
int l=0;
int r=d;
//窗口内的初始值
for (int i=l;i<r;i++){
if (map.containsKey(i)){
List<Integer> list = map.get(i);
for (Integer id : list) {
cnt[id]++;
if (cnt[id]>=k) isOk[id]=true;
}
}
}
//滑动窗口枚举时间
while (r<N){
if (map.containsKey(l)){
for (Integer id : map.get(l)) {
cnt[id]--;
}
}
l++;
r++;
if (map.containsKey(r)) {
for (Integer id : map.get(r)) {
cnt[id]++;
if (cnt[id]>=k) isOk[id]=true;
}
}
}
for (int i=0;i<N;i++){
if (isOk[i]) out.println(i);
}
out.flush();
}
}