题目描述
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Scanner;
public class Main {
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[][] arr = new int[N][2];
int maxts = 0;
HashSet<Integer> HSet = new HashSet<>();
Map<Integer,Integer> map = new HashMap<>();
for(int i=0;i<N;i++){
int ts = sc.nextInt();
int id = sc.nextInt();
arr[i][0] = ts;
arr[i][1] = id;
maxts = maxts>ts?maxts:ts;
}
for(int i=0;i<maxts-D+2;i++){//i用来判断Time区间
for(int j=0;j<N;j++){
if(arr[j][0]>=i && arr[j][0]<i+D){
if(map.containsKey(arr[j][1])){
map.replace(arr[j][1], map.get(arr[j][1])+1);
if(map.get(arr[j][1])==K){
HSet.add(arr[j][1]);
}
}else{
map.put(arr[j][1], 1);
}
}
}
map.clear();
}
int len = HSet.size(),z = 0;
int[] sortArr = new int[len];
Iterator it = HSet.iterator();
while(it.hasNext()){
Object T = it.next();
sortArr[z] = (int)T;
z++;
}
Arrays.sort(sortArr);
for(int o : sortArr){
System.out.println(o);
}
}
}