小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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
代码整体思路是先对数据ts进行排序,
然后使用滑动窗口。
#include<iostream>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
struct node{
int ts,id;
};
bool compare(node &p1,node &p2){
return p1.ts<p2.ts;
}
int main(){
int n,d,k,i;
//m是窗口,t是用来记录符合条件的值
map<int,int> m,t;
cin>>n>>d>>k;
vector<node> v(n);
for(i=0;i<n;i++){
cin>>v[i].ts>>v[i].id;
}
//排序
sort(v.begin(),v.end(),compare);
int l=0,r=0;
while(r<n){
//右移窗口
while(r<n&&v[r].ts-v[l].ts<d){
m[v[r].id]++;
//符合条件 记录id
if(m[v[r].id]>=k){
t[v[r].id]++;
}
r++;
}
//收缩清楚左侧元素
m[v[l].id]--;
l++;
}
//输出
for(map<int,int >::iterator it=t.begin();it!=t.end();it++){
if(it->second>0){
printf("%d\n",it->first);
}
}
return 0;
}