【PAT】B1030 完美数列

最新推荐文章于 2022-07-28 18:46:14 发布

原创最新推荐文章于 2022-07-28 18:46:14 发布 · 327 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

PAT

本文介绍了一种求解完美数列的最大长度的算法。给定一个正整数数列和参数p，若数列中最大值M小于等于最小值m乘以p，则该数列为完美数列。通过排序和迭代计算，可以找出由给定数字构成的最长完美数列。

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

输出样例：

思路：

先把所有数存起来，用 double 防止做乘法的时候爆 int，然后从小到大排序，sort 就可以了。按照这个题的要求，我把每个数当成数列中的最小数，然后算出以这个数为起始的数列的长度，在这里需要注意的就是每一轮计算完成后下一轮计算可以直接使用上一轮计算出的长度 cnt，这样首先是保证了运行速度，其次使 cnt 始终保持是最大值，简化代码。

代码：

#include <cstdio>
#include <algorithm>
using namespace std;

const int maxn = 1e5 + 7;
double a[maxn];

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