原题链接:日志统计
题目描述
小明维护着一个程序员论坛。现在他收集了一份”点赞”日志,日志共有 $N$ 行。
其中每一行的格式是:
ts id
表示在 $ts$ 时刻编号 $id$ 的帖子收到一个”赞”。
现在小明想统计有哪些帖子曾经是”热帖”。
如果一个帖子曾在任意一个长度为 $D$ 的时间段内收到不少于 $K$ 个赞,小明就认为这个帖子曾是”热帖”。
具体来说,如果存在某个时刻 $T$ 满足该帖在$[T,T+D)$ 这段时间内(注意是左闭右开区间)收到不少于 $K$ 个赞,该帖就曾是”热帖”。
给定日志,请你帮助小明统计出所有曾是”热帖”的帖子编号。
输入格式
第一行包含三个整数 $N,D,K。$
以下 $N$ 行每行一条日志,包含两个整数 $ts$ 和 $id$。
输出格式
按从小到大的顺序输出热帖 $id$。
每个 $id$ 占一行。
数据范围
$1≤K≤N≤105,$
$0≤ts,id≤105,$
$1≤D≤10000$
输入样例1:
7 10 2
0 1
0 10
10 10
10 1
9 1
100 3
100 3
输出样例1:
1
3
双指针
先根据时刻排序
因为序列是有序的,所以可以保证区间内的时刻是存在单调性的
所以我们只需要枚举每一个基准点,利用双指针基于该起始点右移,如果在范围内且大于等于$K$个点赞,就存入答案
基准点右移时需要将该基准点的值减一,因为已经移除区间,不必再统计点赞个数。
时间复杂度
$O(nlogn)$
C++ 代码
#include <iostream>
#include <algorithm>
using namespace std;
#define x first
#define y second
typedef pair<int,int> PII;
const int N = 100010;
PII w[N];
int cnt[N];
bool res[N]; // 存放答案
int n, d, k;
int main()
{
cin >> n >> d >> k;
int a, b;
for(int i = 0;i < n;i ++) {
scanf("%d%d",&a,&b);
w[i] = {a,b};
}
sort(w,w + n);
for(int i = 0,j = 0;i < n;i ++) {
while(j < n && w[j].x - w[i].x < d){
cnt[w[j].y] ++;
if(cnt[w[j].y] >= k) res[w[j].y] = true;
j ++;
}
cnt[w[i].y] --;
}
for(int i = 0;i < N;i ++)
if(res[i]) printf("%d\n",i);
return 0;
}