[蓝桥杯] 日志统计

日志统计

题目

小明维护着一个程序员论坛。现在他收集了一份”点赞”日志，日志共有 N 行。

其中每一行的格式是：

ts id
表示在 ts 时刻编号 id 的帖子收到一个”赞”。

现在小明想统计有哪些帖子曾经是”热帖”。

如果一个帖子曾在任意一个长度为 D 的时间段内收到不少于 K 个赞，小明就认为这个帖子曾是”热帖”。

具体来说，如果存在某个时刻 T 满足该帖在 [T,T+D) 这段时间内(注意是左闭右开区间)收到不少于 K 个赞，该帖就曾是”热帖”。

给定日志，请你帮助小明统计出所有曾是”热帖”的帖子编号。

输入格式

第一行包含三个整数 N,D,K。

以下 N 行每行一条日志，包含两个整数 ts 和 id。

输出格式

按从小到大的顺序输出热帖 id。

每个 id 占一行。

数据范围

$$\begin{gathered} 1\le K\le N\le 10^5 \\ 0\le ts,id\le 10^5 \\ 1\le D\le 10000 \end{gathered}$$

输入样例

7 10 2
0 1
0 10
10 10
10 1
9 1
100 3
100 3

输出样例

1
3

题解

思路

• 由于暴力算法存在重复计算相同时间段的情况
• 故选用双指针算法进行优化

#include <iostream> 
#include <cstring>
#include <algorithm>

#define x first
#define y second

using namespace std;
typedef pair<int, int> PII;

const int N = 1e5 + 20;

int n, d, k, cnt[N];
PII p[N];
bool st[N];

int main () {
    cin >> n >> d >> k;
    for (int i = 0; i < n; i ++) cin >> p[i].x >> p[i].y;
    
    sort(p, p + n);
    
    for (int i = 0, j = 0; i < n; i ++) {
        cnt[p[i].y] ++;
        while (p[i].x - p[j].x >= d) {
            cnt[p[j].y] --;
            j ++;
        }
        if (cnt[p[i].y] >= k) st[p[i].y] = true;
    }
    
    for (int i = 0; i <= 1e5; i ++) 
        if (st[i])
            cout << i << endl;    
    return 0;
}