PAT乙级1030 完美数列 测试点4/5超时解决

1030 完美数列 (25分)

给定一个正整数数列，和正整数 p，设这个数列中的最大值是 M，最小值是 m，如果 M≤mp，则称这个数列是完美数列。 现在给定参数 p 和一些正整数，请你从中选择尽可能多的数构成一个完美数列。

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

8

注意数据范围，用long类型，解决测试点5

具体算法参加代码注释

#include <iostream>
#include <stdio.h>
#include <algorithm>

using namespace std;

int main() {
    
    
    long n, p;//定义long类型，解决测试点5
    cin >> n >> p;
    long num[n];
    for(long i = 1; i <= n; i++) {
    
    
        cin >> num[i - 1];
    }
    sort(num, num + n);//自然顺序排序
    long sum = 1;//最后要输出的结果
    for(long i = n; i > sum; i--){
    
    //从后向前遍历，解决测试点4超时
        long a;
        for(a = i - sum; a >= 1; a--) {
    
    
            if(num[i - 1] > p * num[a - 1]) {
    
    //寻找恰好不是完美数列的索引位置
                sum = i - a;
                break;
            }
        }
        if(a == 0) {
    
    //如果上一个循环自然结束，说明本次遍历的元素与它前面所有的数可组成完美数列，已经得到结果了
            sum = i - a;
            break;
        }
    }
    cout << sum << endl;
    return 0;
}