PAT basic-level 1030 完美数列 笔记

1030. 完美数列(25)

时间限制
300 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CAO, Peng
给定一个正整数数列，和正整数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

---

第4个测试点一直运行超时，非常闹心，所以输入输出换成c-方式也不行，想必是故意设了个大数的坑。

• 原本思路是逐个检查长度并且选出最大长度len，循环次数可以不用0到n，用0到n-len可以减少一点循环时间，但也不行。

• 第二次换成了先排序，检测以数组首元素为最小元素的最长序列长度len，然后往后只要检测从len开始的满足<=conq[i]*p的元素个数j，len+j-1就是新的长度，这个时候已经很接近了，但还是没能成功，有点闹心，在网上瞅了瞅别人的思路，代码也懒得看，自己尝试在思路二的基础上修改了一下

• 于是有了思路三，在第一次找到len后，开始循环，每次循环判断conq[j + len + i] 是否<= conq[i] * p，如果不是，那以i元素为最小元素的完美数列长度肯定不是最大长度，可以直接break排除掉

事实上最坏情况下还是O(N^2)的复杂度，但多了这一步就不会超时，估计跟第四个测试用例也有关系…anyhow，贴代码

#include <iostream>
#include <vector>
#include <iterator>
#include <algorithm>
using namespace std;

int main(){
    int n = 0;
    long long p = 0;
    scanf("%d %lld", &n, &p);
    vector<long long> conq(n, 0);
    for (int i = 0; i < n; i++)
        scanf("%lld", &conq[i]);
    sort(conq.begin(), conq.end());
    int len = 0;
    long long  m = conq[0] * p;
    for (int i = 0; i < n; i++){
        if (conq[i] >= conq[0] && conq[i] <= m)
            len++;
    }
    if (len == n){
        printf("%d", len);
        return 0;
    }
    for (int i = 1; i < n; i++){
        int j = 0;
        for (; j + len + i < n && j < n; j++){
            if (conq[j + len + i] <= conq[i] * p)
                continue;
            else
                break;
        }
        len += j;
    }
    printf("%d", len);
    return 0;
}