[PAT乙级]1030. 完美数列(25)

1030. 完美数列(25)

原题链接

给定一个正整数数列，和正整数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

注意：

• 注意数据范围，定义为long long,防止相乘数据溢出
• 点击查看参考链接

代码：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int cmp(int a, int b){return a < b;}
int getMax(int a, int b){return a>b ? a : b;}
int main()
{
    int n;
    long long p;
    cin >> n >> p;
    if(n == 0){
        cout << n;
        return 0;
    }
    vector<long long int> arr(n);
    for(int i=0; i<n; i++)
        cin >> arr[i];
    sort(arr.begin(), arr.end(), cmp);
    int res=1;
    for(int i=0; i<n-1; i++){ //遍历，将arr[i]作为最小数
        for(int j=i+res; j<n; j++){//j置为要满足可以更新数列长度的值，减少循环次数
            if(arr[i]*p < arr[j])//如果不满足条件了，则将下一个元素arr[i+1]最为最小值
                break;
            res = getMax(res, j-i+1);//如果此次的长度大于上一次，更新数列长度
        }
    }
    cout << res << endl;
    return 0;
}