Problem
小明维护着一个程序员论坛。现在他收集了一份"点赞"日志,日志共有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
Process
当看到任一区间的时候我就觉得可能要用尺取法了,不然暴力求解会很麻烦,数据量又那么大。
Code
#include<bits/stdc++.h>
using namespace std;
#define MAXN 100010
map<int, vector<int>>m;
map<int, vector<int>>::iterator it;
vector<int> vs;
int vis[MAXN] = { 0 };
int n, d, k;
bool judge(vector<int> s) {
int low = 0, high = 0, sum = 0, len = s.size();
sort(s.begin(), s.end());
while (low<=high&&high<len) {
sum++;
if (sum >= k)
if (s[high] - s[low] < d)
return true;
else
{
sum--;
low++;
}
high++;
}
return false;
}
int main()
{
cin >> n >> d >> k;
for (int i = 0; i < n; i++) {
int ts, id;
scanf("%d %d", &ts, &id);
m[id].push_back(ts);
}
for (it = m.begin(); it != m.end(); it++)
if (judge(it->second))
vs.push_back(it->first);
for (int i = 0; i < vs.size(); i++)
printf("%d\n", vs[i]);
return 0;
}