标题:日志统计
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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号是否符合题目要求。
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++;
}
}
return false;
}
}