[藍橋杯] 日誌統計

日誌統計

題目

小明維護着一個程序員論壇。現在他收集了一份”點贊”日誌，日誌共有 N 行。

其中每一行的格式是：

ts id
表示在 ts 時刻編號 id 的帖子收到一個”贊”。

現在小明想統計有哪些帖子曾經是”熱帖”。

如果一個帖子曾在任意一個長度爲 D 的時間段內收到不少於 K 個贊，小明就認爲這個帖子曾是”熱帖”。

具體來說，如果存在某個時刻 T 滿足該帖在 [T,T+D) 這段時間內(注意是左閉右開區間)收到不少於 K 個贊，該帖就曾是”熱帖”。

給定日誌，請你幫助小明統計出所有曾是”熱帖”的帖子編號。

輸入格式

第一行包含三個整數 N,D,K。

以下 N 行每行一條日誌，包含兩個整數 ts 和 id。

輸出格式

按從小到大的順序輸出熱帖 id。

每個 id 佔一行。

數據範圍

$1≤K≤N≤10^5 \\ 0≤ts,id≤10^5\\ 1≤D≤10000$

輸入樣例

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;
}