给定一个正整数数列,和正整数 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
思路1:二分法
简化版:
看到最大最小值->排序->有个结论:选出的数个数最大的方案,一定是在连续的递增序列中选出连续的若干个数的方案->强行遍历会超时->对每一个a[i]采用二分法查找的做法(采用upper_bound函数更简洁),具体做法见红色五角星
知识点补充”:
upper_bound()的使用
第一个参数:数组起始位置的地址,注意是内存地址,不是逻辑上的地址,一般要加上数组首地址
第二个参数:数组结束位置的地址
注意:前两个参数区间是左闭右开
第三个参数x以及返回值:返回第一个大于x的地址
还有一个函数作为区分lower_bound():唯一的区别就是返回第一个大于等于x的地址,有个等于!这两个函数一起使用有一个好处:两个返回值相减等于某的连续递增序列中所有值等于x的个数
AC代码:
//#include<iostream>
//#include<cstring>
#include<algorithm>
#include<cstdio>
//#include<cmath>
using namespace std;
const int maxn=100010;
int main()
{
int a[maxn],p,N;
scanf("%d%d",&N,&p);
for(int i=0;i<N;i++){
scanf("%d",&a[i]);
}
sort(a,a+N); //递增排序
int ans=1; //最大长度,初始值为1(至少要有一个)
for(int i=0;i<N;i++){
//对于每个a[i],在a[i+1]~a[N-1]中找到第一个大于a[i]*p的下标
int j=upper_bound(a+i+1,a+N,(long long)a[i]*p)-a;
ans=max(ans,j-i); //j-i刚好是满足M<=p*m数的个数
}
printf("%d\n",ans);
return 0;
}
思路二:two pointers
//#include<iostream>
//#include<cstring>
#include<algorithm>
#include<cstdio>
//#include<cmath>
using namespace std;
const int maxn=100010;
int main()
{
int a[maxn],p,N;
scanf("%d%d",&N,&p);
for(int i=0;i<N;i++){
scanf("%d",&a[i]);
}
sort(a,a+N); //递增排序
int count=1,i=0,j=0;
while(i<N&&j<N){
//j不断右移,直到不满足条件
while(j<N&&a[j]<=(long long)a[i]*p){
count=max(count,j-i+1);
j++;
}
i++;
}
printf("%d\n",count);
return 0;
}
两种方案都是在递增序列去找刚好不满足条件的最大长度
总结:
- a[i]*p计算时只要先将其中一个数先转化为long再计算,那么最后结果就是long long,而long long足够存储;而如果先整体计算再转化,那么结果还是会溢出,可以这么理解 a[i]*p两个都是int计算结果还是int,计算完就需要int存放,显然是存放不了的,那么再转化为long long也是将int所能存放的转化了过去,并不是我们想要的结果
- 学习了个结论:选出的数个数最大的方案,一定是在连续的递增序列中选出连续的若干个数的方案,记住它
- 学习了low_bound()和upper_bound函数的使用