原题链接:日志统计
小明维护着一个程序员论坛。现在他收集了一份”点赞”日志,日志共有 N N N 行。
其中每一行的格式是:
ts id
表示在 t s ts ts 时刻编号 i d id id 的帖子收到一个”赞”。
现在小明想统计有哪些帖子曾经是”热帖”。
如果一个帖子曾在任意一个长度为 D D D 的时间段内收到不少于 K K K 个赞,小明就认为这个帖子曾是”热帖”。
具体来说,如果存在某个时刻 T T T 满足该帖在 [ T , T + D ) [T,T+D) [T,T+D) 这段时间内(注意是左闭右开区间)收到不少于 K K K 个赞,该帖就曾是”热帖”。
给定日志,请你帮助小明统计出所有曾是”热帖”的帖子编号。
第一行包含三个整数 N , D , K 。 N,D,K。 N,D,K。
以下 N N N 行每行一条日志,包含两个整数 t s ts ts 和 i d id id。
按从小到大的顺序输出热帖 i d id id。
每个 i d id id 占一行。
1 ≤ K ≤ N ≤ 105 , 1≤K≤N≤105, 1≤K≤N≤105,
0 ≤ t s , i d ≤ 105 , 0≤ts,id≤105, 0≤ts,id≤105,
1 ≤ D ≤ 10000 1≤D≤10000 1≤D≤10000
7 10 2
0 1
0 10
10 10
10 1
9 1
100 3
100 3
1
3
先根据时刻排序
因为序列是有序的,所以可以保证区间内的时刻是存在单调性的
所以我们只需要枚举每一个基准点,利用双指针基于该起始点右移,如果在范围内且大于等于 K K K个点赞,就存入答案
基准点右移时需要将该基准点的值减一,因为已经移除区间,不必再统计点赞个数。
O ( n l o g n ) O(nlogn) O(nlogn)
#include
#include
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;
}