PAT 1030 完美数列

https://pintia.cn/problem-sets/994805260223102976/problems/994805291311284224

 

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

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

8

时间复杂度：$O(N ^ 2 / 2)$

代码：

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

const int maxn = 1e5 + 10;
long long a[maxn];

int main() {
    int N, p, temp = 1;
    scanf("%d%d", &N, &p);
    for(int i = 1; i <= N; i ++)
        scanf("%lld", &a[i]);
    sort(a + 1, a + 1 + N);
    long long sum = a[1] * p;
    for(int i = 1;i <= N; i ++) {
        for(int j = temp + i; j <= N;j ++) {
            if(a[j] <= a[i] * p)
                temp = j - i + 1;
            else break;
        }
    }
    printf("%d\n", temp);
    return 0;
}