原题
小明维护着一个程序员论坛。现在他收集了一份“点赞”日志,日志共有 NN 行。其中每一行的格式是 ts id
,表示在 tst时刻编号 id 的帖子收到一个“赞”。
现在小明想统计有哪些帖子曾经是“热帖”。如果一个帖子曾在任意一个长度为 D 的时间段内收到不少于 K个赞,小明就认为这个帖子曾是“热帖”。
具体来说,如果存在某个时刻 TT 满足该帖在 [T,T+D)这段时间内(注意是左闭右开区间)收到不少于 K 个赞,该帖就曾是“热帖”。
给定日志,请你帮助小明统计出所有曾是“热帖”的帖子编号。
据大佬的讲解:
这个题其实就是一个排序加典型的双指针,在排完序后遍历数组一次最多遍历长度为d 的区间,最后在 循环末尾更新i ,当然一个帖子有可能多次成为热帖,因此我们用 set来对答案去重,同时 set会自动对存入的数从小到大排序。 在程序中我们遍历时的需求是连续的点赞是属于同一个帖子且点赞时间从早到晚,因此我们用pair<int,int>来存放点赞信息,在我们写排序的cmp时先比较是否是同一个帖子,帖子编号小 的在前,若是同一个帖子再比较时间早晚,时间早的在前。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
bool cmp(PII a,PII b)
{
if(a.second != b.second)
return a.second < b.second;
else return a.first < b.first;
}
void solve()
{
int n,d,k;
cin >> n >> d >> k;
vector<PII> q(n);
set<int> res;
for(int i = 0; i < n; i ++)
cin >> q[i].first >> q[i].second;
sort(q.begin(),q.end(),cmp);
for(int i = 0; i < n; i ++)
{
int j = i;
while(q[i].second == q[j].second && q[j].first < q[i].first + d) j ++;
if(j - i >= k)
res.insert(q[i].second),i = -- j;
}
for(auto it : res)
cout << it << endl;
return;
}
int main()
{
// IOS;
int t = 1;
// cin >> t;
while(t --)
solve();
return 0;
}