PTA:B1030/A1085 完美数列。c++

完美数列

题面：

给定一个正整数数列，和正整数 p，设这个数列中的最大值是 M，最小值是 m，如果 M≤mp，则称这个数列是完美数列。
现在给定参数 p 和一些正整数，请你从中选择尽可能多的数构成一个完美数列。

输入第一行给出两个正整数 N 和 p，其中 N（≤10⁵​​）是输入的正整数的个数，p（≤10​⁹​​）是给定的参数。第二行给出 N 个正整数，每个数不超过 10​⁹​​。

在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。

sample input:

10 8
2 3 20 4 5 1 6 7 8 9

sample output:

8

思路：

• 因为考虑到选出序列的最大值和最小值，所以首先将输入的序列进行排序，方便之后的处理
• 在一个给定的递增序列中找出左边界和右边界，保证满足左端点值不超过右端点值*p，并且满足右端点与左端点的距离最大
• 可以使用二重循环进行枚举，但是给定的参数的值比较大，如果使用循环会爆掉，同时也要注意，使用int也会超，要用long long 来表示乘法之后的结果
• 使用二分查找来降低时间复杂度，由于是递增序列，所以如果改位置的值比预想的值小，说明下一次要在mid的后边进行查找，如果比预想值大，那就在mid的前半段进行查找
• 代码中的search函数就是寻找符合要求的值的位置的函数，也可以使用upper_bound函数来代替，这样可以减少代码量

AC代码

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
using namespace std;
int a[100001]; 
int n;
int search(int i,long long x)//对于数组寻找第一个大于x的位置 
{
 int l=i+1,r=n-1,mid=0; 
 if(a[n-1]<=x) return n;//所有数都不大于
 while(l<r)
 {
  mid=(l+r)/2;
  if(a[mid]<=x)//说明大于mid的肯定在右边/后边
   l=mid+1;
  else r=mid; 
  } 
 return l;
}
int main()
{
 int p;
 cin>>n>>p;
 for(int i=0;i<n;i++)
 cin>>a[i];
 sort(a,a+n);
 int ans=1;
 for(int i=0;i<n;i++)
 {
  //查找第一个超过a[i]*p的数，返回他的位置
  int pos=search(i,(long long)a[i]*p);
  ans=max(ans,pos-i); 
 }
 cout<<ans<<endl;
 return 0;
}