[蓝桥杯2018初赛]日志统计

题目描述
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志，日志共有N行。
其中每一行的格式是：ts id。表示在ts时刻编号id的帖子收到一个"赞"。
现在小明想统计有哪些帖子曾经是"热帖"。
如果一个帖子曾在任意一个长度为D的时间段内收到不少于K个赞，小明就认为这个帖子曾是"热帖"。
具体来说，如果存在某个时刻T满足该帖在[T, T+D)这段时间内(注意是左闭右开区间)收到不少于K个赞，该帖就曾是"热帖"。
给定日志，请你帮助小明统计出所有曾是"热帖"的帖子编号。

输入
第一行包含三个整数N、D和K。
以下N行每行一条日志，包含两个整数ts和id。
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

AC代码

#include<bits/stdc++.h>
using namespace std;

int n,d,k;
const int N=100005;
vector<int> v[N];

bool judge(int i){
	for(int j=0;j<v[i].size();j++){
		if(j+k-1>=v[i].size()) break;
		if(v[i][j+k-1]-v[i][j]<d){
			
			return true;
		}
	}
	return false;
}

int main(){	
	cin>>n>>d>>k;
    map<int,bool> mm;
    
	while(n--){
		int ts,id;
		cin>>ts>>id;
		mm[id]=true;
		v[id].push_back(ts);
	}
	
	for(int i=0;i<N;i++){
		if(!mm[i]) continue;
		
		sort(v[i].begin(),v[i].end());		
		if(judge(i)){
			cout<<i<<endl;
		}
	}
	
	return 0;
}