题目描述
小明维护着一个程序员论坛。现在他收集了一份”点赞”日志,日志共有 N 行。
其中每一行的格式是:
ts id
表示在 ts 时刻编号 id 的帖子收到一个”赞”。
现在小明想统计有哪些帖子曾经是”热帖”。
如果一个帖子曾在任意一个长度为 D 的时间段内收到不少于 K 个赞,小明就认为这个帖子曾是”热帖”。
具体来说,如果存在某个时刻 T 满足该帖在 [T,T+D) 这段时间内(注意是左闭右开区间)收到不少于 K 个赞,该帖就曾是”热帖”。
给定日志,请你帮助小明统计出所有曾是”热帖”的帖子编号。
输入格式
第一行包含三个整数 N,D,K。
以下 N 行每行一条日志,包含两个整数 ts 和 id。
输出格式
按从小到大的顺序输出热帖 id。
每个 id 占一行。
数据范围: 1≤K≤N≤10E5,
0≤ts,id≤10E5,
1≤D≤10000
#include<bits/stdc++.h> //万能头文件
using namespace std;
struct log{
int time;
int id;
}; //定义一个日志的结构体,包含时间和点赞的日志id
bool cmp(struct log l1,struct log l2) //比较函数,规定了按照时间升序排列
{
if(l1.time>l2.time) //前面的数大于后面,则输出false
{
return false;
}
return true; //前面的数小于等于后面,输出true,故是升序排列
}
int main()
{
int N,D,K; //待输入的日志数量N、规定的时间间隔D,点赞数量的阈值K
struct log l[100000];//因为题目中说明了0≤ts,id≤10E5,所以这里设置为最大的日志数量
cin>>N>>D>>K;//输入
int sum[100000],panduan[10000];//sum[i]代表了id为i的日志收获的赞的总和,panduan[i]表示id为i的日志是否为热帖
for(int i=0;i<=10000;i++) //因为id可以等于10000,所以注意循环条件
panduan[i]=0; //这里初始化panduan每一个为0,为0则代表不是热帖
for(int i=0;i<N;i++)
cin>>l[i].time>>l[i].id; //输入点赞时间time和id
sort(l,l+N,cmp); //对于日志l[]数组,按照时间的升序对其进行排列
for(int i=0,j=0;i<N;i++)
{
sum[l[i].id]++; //首先先对日志收获的赞的总和加1
while(l[i].time-l[j].time>=D) //如果目前的第i个日志与第j个日志的时间差距>=D (初始时,j=0,即代表是第一个日志)
{
sum[l[j].id]--;//当时间超出规定的间隔时,则会对第j个日志对应的id的点赞数-1,因为其超出了时间间隔,一定是删除前面的,不能删后面的,因为后面的可能在后续的间隔D中满足热帖要求,而最前面的一定不满足了
j++; //首位的日志往后移,以便其符合在时间间隔内的条件
}
if(sum[l[i].id]>=K) //如果第i个日志对应的id的点赞数和超过了规定的阈值K,则满足热帖要求
panduan[l[i].id]=1; //将其对应的panduan数组设置为1 ,即为热帖
}
for(int i=0;i<sizeof(panduan);i++) //注意这里的循环条件不是N,而是panduan数组的长度,因为热帖的id没有规定在N之内,所以规定为数组长度才不会有遗漏
{
if(panduan[i]==1)
cout<<i<<endl;//注意这里的i代表的不是第i个日志,代表的是日志的id,所以直接输出就好,无需对i进行修改
}
return 0;
}
使用:sort (s1,s2,cmp)函数,可以直接对数组排序,第一个参数是要排序的区间首地址,第二个参数是区间尾地址的下一地址,排序的区间是[a,b)。默认是升序排列。如果是没有定义小于运算的数据类型,或者想改变排序的顺序,就要用到第三参数----比较函数。比较函数是一个自己定义的函数,返回值是 bool 型,它规定了什么样的关系才是“小于”。想把刚才的整数数组按降序排列,可以先定义一个比较函数 cmp
代码中有着详细的标注,可供参考