[蓝桥杯][2018年第九届真题]日志统计(树状数组)

题目描述
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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)/~

发布了652 篇原创文章 · 获赞 101 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/starlet_kiss/article/details/105342841