先对数组进行从小到大排序,对每个元素运用二分查找该数*p+1,得到位置,但此时还要向前减去相同的数。然后再找出最大值
AC代码:
#include<iostream>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<list>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define ll long long
#define inf 999999999
using namespace std;
long long n,p;
long long a[100005];
long long f(long long x)
{
long long l=0,r=n-1;
while(l<=r)
{
long long mid=(l+r)/2;
if(a[mid]>x)
{
r=mid-1;
}
else if(a[mid]<x)
{
l=mid+1;
}
else
{
return mid;
}
}
return l;
}
int main()
{
scanf("%lld %lld",&n,&p);
for(long long i=0;i<n;i++)
{
scanf("%lld",&a[i]);
}
sort(a,a+n);
long long maxx=0;
for(long long i=0;i<n;i++)
{
long long pos=f(a[i]*p+1);
//cout<<pos<<endl;
for(int j=pos-1;;j--)
{
if(a[j]==a[i]*p+1)
{
pos--;
}
else
{
break;
}
}
if(pos-i>maxx)
{
maxx=pos-i;
}
}
printf("%lld",maxx);
}