给定一个正整数数列,和正整数p,设这个数列中的最大值是M,最小值是m,如果M <= m * p,则称这个数列是完美数列。
现在给定参数p和一些正整数,请你从中选择尽可能多的数构成一个完美数列。
输入格式:
输入第一行给出两个正整数N和p,其中N(<= 105)是输入的正整数的个数,p(<= 109)是给定的参数。第二行给出N个正整数,每个数不超过109。
输出格式:
在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。
输入样例:
10 8
2 3 20 4 5 1 6 7 8 9
输出样例:
8
解法一,二分法:
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 100010;
int n, p, a[maxn];
int binarySearch(int i, long long x) {
if (a[n - 1] <= x) {
return n;
}
int l = i - 1, r = n - 1, mid;
while (l < r) {
mid = (l + r) / 2;
if (a[mid] < x) {
l = mid + 1;
} else {
r = mid;
}
}
return l;
}
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 = binarySearch(i, (long long) a[i] * p);
ans = max(ans, j - i);
}
printf("%d\n", ans);
return 0;
}
解法二,two pointers
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 100010;
int a[maxn];
int main() {
int n, p;
int i = 0, j = 0, count = 1;
scanf("%d%d", &n, &p);
for (int i = 0; i < n; i++) {
scanf("%d", &a[i]);
}
sort(a, a + n);
while (i < n && j < n) {
while (j < n && a[j] <= (long long)a[i] * p) {
count = max(count, j - i + 1);
j++;
}
i++;
}
printf("%d\n", count);
return 0;
}