题目描述
小明维护着一个程序员论坛。现在他收集了一份”点赞”日志,日志共有 N行。
其中每一行的格式是:
ts id
表示在 ts时刻编号 id的帖子收到一个”赞”。
现在小明想统计有哪些帖子曾经是”热帖”。
如果一个帖子曾在任意一个长度为 D的时间段内收到不少于 K个赞,小明就认为这个帖子曾是”热帖”。
具体来说,如果存在某个时刻 T
满足该帖在 [T,T+D) 这段时间内(注意是左闭右开区间)收到不少于 K个赞,该帖就曾是”热帖”。
给定日志,请你帮助小明统计出所有曾是”热帖”的帖子编号。
输入格式
第一行包含三个整数 N,D,K。
以下 N行每行一条日志,包含两个整数 ts 和 id。
输出格式
按从小到大的顺序输出热帖 id。
每个 id占一行。
数据范围
1≤K≤N≤105,
0≤ts,id≤105,
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> II;
int n,d,k;
II logs[100005];
bool st[100005];
int main(int argc, char** argv) {
scanf("%d%d%d",&n,&d,&k);
for(int i=0;i<n;i++){
scanf("%d%d",&logs[i].x,&logs[i].y);
}
sort(logs,logs+n);
for(int i=0;i<n;i++){//i在左端。j在右端
int t=logs[i].x;
int cnt=1;
for(int j=i+1;j<n;j++){
if(logs[j].x-t<d&&logs[j].y==logs[i].y){
cnt++;
}
}
if(cnt>=k){
st[logs[i].y]=true;
}
}
for(int i=1;i<=100005;i++){
if(st[i]){
printf("%d\n",i);
}
}
return 0;
}
优化版:
#include <iostream>
#include <algorithm>
#include <utility>
#define II pair<int,int>//代码逐渐y风
#define x first
#define y second
using namespace std;
int n,d,k;
II logs[100005];//n条记录
int cnt[100005];//用来标记一个id号的获赞数
bool st[100005];//用来标记一个id号是否是热号
int main(int argc, char** argv) {
scanf("%d%d%d",&n,&d,&k);
int i,j;
for(i=0;i<n;i++){
scanf("%d%d",&logs[i].x,&logs[i].y);
}
sort(logs,logs+n);
for(i=0,j=0;i<n;i++){//i在右端走,j在左端走,遍历每条记录
int id=logs[i].y;//第
cnt[id]++;
while(logs[i].x-logs[j].x>=d){// 第j条帖子与第i条帖子的时间超过了d,就说明第j条记录的赞要过期了
cnt[logs[j].y]--;//就是这条帖子的
j++;
}
if(cnt[id]>=k){
st[id]=true;
}
}
for(i=0;i<100005;i++){
if(st[i])
printf("%d\n",i);
}
return 0;
}