日志统计（蓝桥杯）

蓝桥杯的一道题：

原题链接：传送门

思路：先说最容易想到的，把ts都放到他的id下，枚举每个id,在每个id之下判断是否存在热帖，存在的条件就是在[t,t+d)中cnt>=k

code:

///AC 6/10 
#include<cstdio>
#include<set>
#include<vector>
using namespace std;
vector<int>a[100005];
set<int>vis;
set<int>hot_id;
vector<int>::iterator it;
vector<int>::iterator it2;
set<int>::iterator it1;
int main()
{       
	int n,d,k,ts,id,cnt=0;
	scanf("%d%d%d",&n,&d,&k);
	for(int i=0;i<n;i++) 
	{
		scanf("%d%d",&ts,&id);
		a[id].push_back(ts);//每个id中保存被赞的ts 
		vis.insert(id);//记录id方便后边的枚举 
	}
	for(it1=vis.begin();it1!=vis.end();it1++)//枚举每个id是否是热帖 
	{
		int temp=*it1;
		for(it=a[temp].begin();it!=a[temp].end();it++)
		{
			int l=*it;//定义区间 
			int r=l+d;
			cnt=0; 
			for(it2=a[temp].begin();it2!=a[temp].end();it2++)
			{
				int t=*it2;
				if(t>=l&&t<r)
				cnt++;
			}
			if(cnt>=k)
			{
				hot_id.insert(temp);
				break;
			}
		}
	}
	for(it1=hot_id.begin();it1!=hot_id.end();it1++)
	printf("%d\n",*it1);
	return 0;
 } 
 //代码的效率太低，时间复杂度O（n^3）tle了 

结果悲凉tle了

换种思路：枚举每个id取尺法判断是否有热帖，取尺法感觉不太好想（取尺之前先要排序）

#include<cstdio>
#include<algorithm>
#include<set>
#include<vector>
using namespace std;
int n,d,k,ts,id;
vector<int>a[100005];
set<int>vis,hot_id;
set<int>::iterator it;
int judge(int id)//取尺不好想 
{
	int l=0,r=0,cnt=0;
	sort(a[id].begin(),a[id].end());//取尺之前先排序每个id下的时间 
	while(l<=r&&r<a[id].size()) 
	{
		cnt++;
		if(cnt>=k)
		{
			if(a[id][r]-a[id][l]>=d)
			{
				l++;
				cnt--;
			}
			else
			return 1;
		}
		r++;
	}
	return 0;
}
int main()
{
	scanf("%d%d%d",&n,&d,&k);
	for(int i=0;i<n;i++)
	{
		scanf("%d%d",&ts,&id);
		a[id].push_back(ts);
		vis.insert(id);//保存每个id 
	}
	for(it=vis.begin();it!=vis.end();it++)//枚举每个id 
	{
		if(judge(*it))//热帖 
		printf("%d\n",*it);
	}
	return 0;
 }