给定一个正整数数列,和正整数 p,设这个数列中的最大值是 M,最小值是 m,如果 M≤mp,则称这个数列是完美数列。
现在给定参数 p 和一些正整数,请你从中选择尽可能多的数构成一个完美数列。
输入格式:
输入第一行给出两个正整数 N 和 p,其中 N(≤10^5)是输入的正整数的个数,p(≤10e9)是给定的参数。第二行给出 N 个正整数,每个数不超过 10^9
。
输出格式:
在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。
输入样例:
10 8
2 3 20 4 5 1 6 7 8 9
输出样例:
8
思路:
开一个数组a[],从小到大排序,确定一个左端点a[i]和右端点a[j],使得a[j]<=a[i]*p成立,且j-i最大。
从左至右扫描,对于每一个a[i],在a[i]~a[n-1]内二分查找第一个超过a[i]*p的数的位置j,这样 j-i 就是对a[i]来说满足a[i]*p的最大长度,然后所有 j-i 的最大值
注意:
题目所给数据规模来看,O(n2)肯定会超时(这也是使用二分查找的原因)。同时a[i]*p超出了int的数据范围,所以使用long long数据类型。
样例解释:
10 8
2 3 20 4 5 1 6 7 8 9
排完序后,初始时刻,i==0,对于a[i] (也就是1),在a[1]~a[9]中寻找第一个大于8的数(也就是9)的位置 j,此时j == 8,j - i == 8
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 100010;
int n, p, a[maxn];
//从a[i+1]~a[n-1]中寻找第一个大于x的数的位置
int binarySearch(int i , long long x)
{
if(a[n - 1] <= x) return n; //所有数都不大于x,返回n
int left = i + 1, right = n - 1, mid;
while(left < right)
{
mid = (left + right) / 2;
if(a[mid] <= x)
left = mid + 1;
else
right = mid - 1;
}
return left; //由于结束时left==right,因此返回left和right都可以
}
int main()
{
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++)
{
int j = binarySearch(i, (long long)a[i] * p);
ans = max(ans, j - i); //更新最大长度
}
printf("%d\n",ans);
return 0;
}
使用upper_bound函数,会更加简洁。
lower_bound( )和upper_bound( )都是利用二分查找的方法在一个排好序的数组中进行查找的。
在从小到大的排序数组中,
lower_bound( begin,end,num):从数组的begin位置到end-1位置二分查找第一个大于或等于num的数字,找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。
upper_bound( begin,end,num):从数组的begin位置到end-1位置二分查找第一个大于num的数字,找到返回该数字的地址,不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 100010;
int n, p, a[maxn];
int main()
{
scanf("%d%d", &n, &p);
for(int i = 0; i < n; i++)
scanf("%d", &a[i]);
sort(a, a + n);
int ans = 1;
for(int i = 0; i < n; i++)
{
int j = upper_bound(a + i + 1, a + n, (long long)a[i] * p) - a; //注意upper_bound函数是左闭右开,返回的是地址,两个地址直接相减获取的是两个地址之间的距离
ans = max(ans, j - i);
}
printf("%d", ans);
return 0;
}