返回目录
题意
从N个正整数中选择若干个数,使得选出的这些数中的最大值不超过最小值的p倍。问满足条件的选择方案中,选出的数的最大个数。
样例(可复制)
10 8
2 3 20 4 5 1 6 7 8 9
样例输出
8
提供易错样例
3 3
1 2 3
//output 3
注意点
- 由于题干中涉及序列的最大值和最小值,因此不妨先将所有N个正整数从小到大进行排序。在此基础上证明:能使选出的数个数最大的方案,一定是在该递增序列中选择连续的若干个数的方案。
- p与序列中的元素均可能达到10**9,因此a[i] *p可能达到10的18次方, 所以必须使用long long类型
- 如果强制进行O(n**2)的二重循环枚举,那么根据题目的数据范围,肯定是会超时的。以下用两种方法来解决这个问题:二分查找和two pointers。
- 二分法里用了upper_bound(),该函数底层是使用二分法实现的,具体到本题会在a[i+1]~a[n-1]中查找第一个超过a[i]*p的数,返回其位置指针。也可以自己写个二分查找函数。
- 根据实际测试,两者的时间复杂度基本为线性,相比之下,two pointers略快一点
二分法
#include <bits/stdc++.h>
using namespace std;
int a[100010];
int main(){
int count=1,n,p;
scanf("%d%d",&n,&p);
for(int i=0;i<n;i++)scanf("%d",&a[i]);
sort(a,a+n);
for(int i=0;i<n;i++){
int j=upper_bound(a+i+1,a+n,(long long)a[i]*p)-a;
count=max(count,j-i);
}
printf("%d",count);
return 0;
}
two pointers
#include <bits/stdc++.h>
using namespace std;
int a[100010];
int main(){
int count=0,n,p;
scanf("%d%d",&n,&p);
for(int i=0;i<n;i++)scanf("%d",&a[i]);
sort(a,a+n);
int i,j=0;
for(i=0;i<n;i++){
while(j<n&&(long long)a[j]<=(long long)a[i]*p){
j++;
}
count=max(count,j-i);
}
printf("%d",count);
return 0;
}