题目描述
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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
解题思路
本题可用尺取法,对于对于同一个id的获赞时刻ts进行从小到大排序,通过l,r.从第一个ts时刻开始。当点赞数大于等于k时,判断间隔是否在D内时间段。不满足l++。点赞数小于k,则r往后移动。
直接上代码
#include <bits/stdc++.h>
using namespace std;
const int MAx = 1e5 + 1;
vector<int> a[MAx];
int n,d,k;
bool check(int x) {
int len = a[x].size();//id对应的总点赞数
if (len < k) return false;//总点赞数少于k,不满足
int l = 0,r = 0,sum = 0;
sort(a[x].begin(),a[x].end());
while (l <= r && r < len) {
sum++;
if (sum >= k) {
if (a[x][r] - a[x][l] < d) {
return true;
} else {
sum--;
l++;
}
}
r++; //不够k,继续后移
}
return false;
}
int main() {
cin >> n >> d >> k;
int ts,id;
for (int i = 0;i < n;i++) {
scanf("%d%d",&ts,&id);
a[id].push_back(ts);
}
for (int i = 0;i < MAx;i++) {
if (check(i)) {
printf("%d\n",i);
}
}
return 0;
}