蓝桥杯:日志统计 尺取法
题目描述
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有N行。其中每一行的格式是:
ts id
表示在ts时刻编号id的帖子收到一个"赞"。
现在小明想统计有哪些帖子曾经是"热帖"。如果一个帖子曾在任意一个长度为D的时间段内收到不少于K个赞,小明就认为这个帖子曾是"热帖"。
具体来说,如果存在某个时刻T满足该帖在[T, T+D)这段时间内(注意是左闭右开区间)收到不少于K个赞,该帖就曾是"热帖"。
给定日志,请你帮助小明统计出所有曾是"热帖"的帖子编号。
【输入格式】
第一行包含三个整数N、D和K。
以下N行每行一条日志,包含两个整数ts和id。
【输出格式】
按从小到大的顺序输出热帖id。每个id一行。
【输入样例】
7 10 2
0 1
0 10
10 10
10 1
9 1
100 3
100 3
【输出样例】
1
3
思路
- 二维数组,读取点赞时间ts,id,
a[id][ts]
- 遍历所有id,对每一个id的所有点赞时间升序排序
- 两个指针指向区间的左右
l, r
,然后逐步向右推进,点赞数目就是r-l+1
,相差时间就是a[id][r] - a[id][l]
- 只要找到一个相差时间小于d的,break
这里因为区间长度表示点赞数目,我们只关注点赞数目是否超过阈值,不关心最大最小值,于是区间长度几乎是固定的,尺取就变成一个定长的滑动窗口了。。。。
代码
#include<bits/stdc++.h>
using namespace std;
#define MAXLEN 100009
vector<int> a[MAXLEN];
vector<int> result;
int n, d, k, ts, id;
int main()
{
cin>>n>>d>>k;
for(int i=0; i<n; i++)
{
cin>>ts>>id;
a[id].push_back(ts);
}
for(int i=0; i<MAXLEN; i++)
{
// 点赞时间从小到大排序
sort(a[i].begin(), a[i].end());
// l,r指针指向开头,开始尺取
int l = 0;
int r = 0;
while(l<=r && r<a[i].size())
{
// 赞数够多,时间区间也满足,直接存结果
if(r-l+1>=k && a[i][r]-a[i][l]<d)
{
result.push_back(i);
break;
}
// 赞数够多,但是时间区间不满足,左指针向右压缩
else if(r-l+1>=k && a[i][r]-a[i][l]>=d)
{
l += 1;
}
// 右指针向右推进
r += 1;
}
}
for(int i=0; i<result.size(); i++)
{
cout<<result[i]<<endl;
}
return 0;
}