1030 完美数列 (25分)
给定一个正整数数列,和正整数 p,设这个数列中的最大值是 M,最小值是 m,如果 M≤mp,则称这个数列是完美数列。现在给定参数 p 和一些正整数,请你从中选择尽可能多的数构成一个完美数列。
输入格式:
输入第一行给出两个正整数 N 和 p,其中 N(≤10^5 )是输入的正整数的个数,p(≤10^9 )是给定的参数。第二行给出 N 个正整数,每个数不超过 10^9 。
输出格式:
在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。
输入样例:
10 8
2 3 20 4 5 1 6 7 8 9
输出样例:
8
解题思路:
将数列按递增顺序排列,从第一个最小数开始作为m找到第满足条件的M求出完美数列的最大数max。但是数列长度为10^5,
O(N^2)可能会超时(该题要求200ms)。于是我选择二分查找法的lower_bound找到一个满足大于等于x的位置将时间复杂度降低到NlogN。
AC代码
#include <cstdio>
#include <algorithm>
#include <math.h>
using namespace std;
typedef long long LL;
LL lowerbound(LL A[],LL left,LL right,LL x){
LL mid;
while(left<right){
mid=(left+right)/2;
if(x<=A[mid])
right=mid;
else
left=mid+1;
}
return left;
}
int main(){
LL n,p;
scanf("%ld%ld",&n,&p);
LL num[n];
for(LL i=0;i<n;i++){
scanf("%ld",&num[i]);
}
sort(num,num+n);
LL MAX=0;
for(LL i=0;i<n;i++){
LL res=num[i]*p;
LL id=lowerbound(num,0,n-1,res);
if(num[id]>res)
id--;
if(MAX<id-i+1)
MAX=id-i+1;
}
printf("%d\n",MAX);
return 0;
}