题目描述
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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
思路:将每一个帖子出现的时间保存起来,将每一个时间更新到树状数组中,然后遍历每一个时间,判断这个时间之后d时间内的帖子个数。这道题目有一个地方,就是d的大小不知道,因此d有可能很大,有可能超出界限,如果超出界限直接取最大值再计算就好了。(超出界限有可能爆内存,因为这个错了好多次,yyy)
代码如下:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxx=1e5+100;
int c[maxx];
vector<int> p[maxx];
int n,d,k;
inline int lowbit(int x){return x&-x;}
inline void update(int x,int v)
{
while(x<maxx)
{
c[x]+=v;
x+=lowbit(x);
}
}
inline ll query(int x)
{
ll ans=0;
while(x)
{
ans+=(ll)c[x];
x-=lowbit(x);
}
return ans;
}
int main()
{
scanf("%d%d%d",&n,&d,&k);
int ts,id;
memset(c,0,sizeof(c));
set<int> s;s.clear();
for(int i=1;i<=n;i++)
{
scanf("%d%d",&ts,&id);
s.insert(id);
p[id].push_back(ts+1);
}
vector<int> ans;ans.clear();
for(set<int> ::iterator it=s.begin();it!=s.end();it++)
{
id=*it;
for(int i=0;i<p[id].size();i++) update(p[id][i],1);
for(int i=0;i<p[id].size();i++)
{
ll zz=query(min(p[id][i]+d-1,maxx-1))-query(p[id][i]-1);
if(zz>=k)
{
ans.push_back(id);
break;
}
}
for(int i=0;i<p[id].size();i++) update(p[id][i],-1);
}
sort(ans.begin(),ans.end());
for(int i=0;i<ans.size();i++) printf("%d\n",ans[i]);
return 0;
}
努力加油a啊,(o)/~