尺取法
一开始程序陷入死循环,没想明白。。。虽然它少操作了不少if语句节省时间,但是边界真的不好处理
这样也是对的
while(l<len)
{
while(r<len&&cnt<k)
{
++cnt;
++r;
}
if(cnt<k) break;
else if(cnt>=k)
{
if(g[id][r-1]-g[id][l]<d) return true;
else
{
l++;
cnt--;
}
}
}
#include<iostream>
#include<set>
#include<map>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e5+10;
vector<int> g[N]; //其实用multiset比较好,但是不想写迭代器
vector<int> ans;
int n,d,k;
bool judge(int id)
{
int cnt=0;
sort(g[id].begin(),g[id].end());
int len=g[id].size();
if(len<k) return false;
int l=0,r=0;
while(r<len&&l<=r) //while(i<=r&&i<len)不行
{
while(r<len&&cnt<k)
{
++cnt;
++r; //这边r++可能到len了,while(i<=r&&i<len)卡不住它 于是乎第二个while进不去,if也进不去,死循环
}
if(cnt>=k)
{
if(g[id][r-1]-g[id][l]<d) return true;
else
{
l++;
cnt--;
}
}
}
return false;
// while(l<= r&&r< len)
// {
// sum++;
// if(sum>= k)
// {
// if(g[id][r]-g[id][l]< d)//注意是小于
// return 1;
// else
// l++,sum--;
// }
// r++;
// }
// return false;
}
int main()
{
cin>>n>>d>>k;
int mx=0;
for(int i=1;i<=n;++i)
{
int ts,id;
cin>>ts>>id;
g[id].push_back(ts);
mx=max(mx,id);
}
for(int i=0;i<=mx;++i)
{
if(judge(i))
ans.push_back(i);
}
int sum=ans.size();
for(int i=0;i<sum;++i)
{
cout<<ans[i]<<endl;
}
return 0;
}