这道题一看就知道,是采用二分查找的思路完成。观察题目,可以确定,我们要求的答案(最大整数个数)可以作为二分查找的对象,因为它的值满足:如果当前的答案满足条件,那么最优的答案一定大于或等于当前答案;若不满足条件,那么最优的答案一定小于当前答案。
首先对全部数据进行排序,然后初始化l=1,r=n,进行二分求解。判断mid是否满足条件采用O(n)直接遍历,假设每一个数字均是最小值,确定对应的最大值(假设最小值下标为i,最大值下标一定为i+mid-1),并且判断二者是否满足M<=m*p这个条件,如果满足,直接返回true,若全部数据均不满足,返回false。
一开始,我提交之后,只过了3个点,其余点均超时。我一开始不太理解,我的时间复杂度并没有超过应有的范围。后来,我发现,我的二分答案过程容易陷入死循环,当r=l+1的时候,mid=l,所以新的l依然等于旧的l,新的r也如此,造成了死循环,因此在r=l+1的时候加入特判,直接跳出循环输出答案。
下面是代码:
#include <iostream>
#include<vector>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<map>
#include<cstdlib>
using namespace std;
long long n,p;
long long a[100010];
bool check(long long mid)
{
int w=(int)(n/2);
for(int i=w;i+mid-1<=n;i++)
{
long long M,m;
m=a[i];
M=a[i+mid-1];
if(M<=m*p)return true;
}
for(int i=1;i<=w;i++)
{
if(i+mid-1>n)break;
long long M,m;
m=a[i];
M=a[i+mid-1];
if(M<=m*p)return true;
}
return false;
}
int main() {
scanf("%lld%lld",&n,&p);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
sort(a+1,a+n+1);
long long l,r,mid;
l=1;r=n;
while(l<r)
{
mid=(l+r)>>1;
if(r-l==1)
{
if(check(r))
{
l=r;
break;
}
if(!check(r))
{
break;
}
}
if(check(mid))
{
l=mid;
}
else{
r=mid-1;
}
}
printf("%lld",l);
}