PAT-BASIC-1030-完美数列

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

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

8

排序后，遍历数组的m，M通过二分查找获得,时间复杂度是O(NLogN)
LONG LONG这个地方又天真了....

#include <bits/stdc++.h>
#define LL long long
using namespace std;

LL n, p;
vector<LL> arr;
bool cmp(const LL &a, const LL &b){
    return a < b;
}
int  main(){
    LL res = 1;
    scanf("%lld%lld", &n, &p);
    LL tmp;
    for(int i = 0; i < n; ++i){
        scanf("%lld", &tmp);
        arr.push_back(tmp);
    }
    sort(arr.begin(), arr.end(), cmp);
    for(int i = 0; i < n; ++i){
        LL pos = upper_bound(arr.begin(), arr.end(), arr[i]*p) - arr.begin();
        if(pos != -1){
            res = max(pos-i, res);
        }
        else{
            res = max(n-i, res);
            break;
        }
    }
    printf("%lld\n", res);
    return 0;
}