PAT乙级 1030 完美数列

一、题目

问题描述

给定一个正整数数列，和正整数 p，设这个数列中的最大值是 M，最小值是 m，如果 M≤mp，则称这个数列是完美数列。

现在给定参数 p 和一些正整数，请你从中选择尽可能多的数构成一个完美数列。

输入格式：

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

输出格式：

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

输入样例：

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

输出样例： 

8

二、思路详解

显然先对数据排好序，单纯暴力法不可通过，为把时间复杂度降至O(nlogn)，首要想到的思路是用二分搜索解决。

三、代码实现——二分搜索、双指针

以下是二分搜索代码：

#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN = 1e5 + 5;
int arr[MAXN];
int BinarySearch(int i, int n, int p) {
	int l = i, r = n - 1;
	while (l < r) {
		int mid = l + (r - l + 1) / 2;
		if (arr[mid] <= (long long)arr[i] * p) l = mid;
		else r = mid - 1;
	}
	return l;
}
int main() {
	int n, p;
	cin >> n >> p;
	for (int i = 0; i < n; i++) cin >> arr[i];
	sort(arr, arr + n);
	int cnt = 0;
	for (int i = 0; i < n; i++) {
		int j = BinarySearch(i, n, p);
		cnt = max(cnt, j - i + 1);
	}
	cout << cnt;
	return 0;

}

双指针做法：

#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN = 1e5 + 5;
int arr[MAXN];
int main() {
	int n, p;
	cin >> n >> p;
	for (int i = 0; i < n; i++) cin >> arr[i];
	sort(arr, arr + n);
	int cnt = 0, j = 0, ans = 0;
	for (int i = 0; i < n; i++) {
		while (j < n) {
			if (arr[j] <= (long long)arr[i] * p) {
				cnt++;
				j++;
			}
			else break;
		}
		ans = max(ans, cnt);
		if (ans == n - i) break;
		cnt--;
	}
	cout << ans;
	return 0;
}