PAT乙级-B1030 完美数列（25）【二分法】

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

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

8

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> a;
//使用二分查找法，令i，j = 0，令开始从0～n-1处二分查找出第一个a[i] * p < a[j]的位置。
//此时以a[i]开头的最大序列长度为j - i， 令cnt记录最长长度，并更新。
//再令i + 1， 从j ～ n -1 处找出 a[i+1]的最长序列长度，直至j == n为止。
int main(){
    int n,p;
    cin >> n >> p;
    a.resize(n);
    for(int i = 0 ; i < n; ++i)
        cin >> a[i];
    sort(a.begin(), a.end());
    int cnt = 0, i = 0, j = 0;
    while(j < n){
        int r = n;//由于n - 1处也要被考虑，故，令r = n
        while(j < r){
            int mid = (j + r) / 2;
            if((long long)a[i] * p >= a[mid]) j = mid + 1; //测试点5，a[i]*p > int，要用长整型
            else r = mid;
        }
        cnt = max(cnt, j - i);
        i++;
    }
    cout << cnt << endl;
}