PAT乙级题目1030 完美数列 (笔记)

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

测试点四一直运行超时的代码

//思路：采用比较的方法，首先进行排序然后从第一个数开始作为要构建的完美数列中最小的数m。然后在剩下的数列中找尽可能大的数，且要满足>=mp作为该完美数列中最大的数。然后在给定的数列中遍历寻找<=M&&>=m的数字，记录其个数。这里要定义一个变量作为当前构建完美数列中数字最多的个数。在不断定义m时得到的完美数列中的个数进行比较。
#include<iostream>
int main()
{
    using namespace std;
    int m,M=0,max=0,sum=0,i=0,j=0,book=0;
    long long N,p,a[100005];//如果用int类型测试点5就会答案错误,因在运算a[i]*p的时候会超出int范围的
    cin>>N>>p;
    book=N-1;
    j=N-1;
    for(int i=0;i<N;i++)
        cin>>a[i];
    for(int i=0;i<N;i++){
        int k;
        for(int j=i+1;j<N;j++){
            if(a[i]>a[j]){
                k=a[i];a[i]=a[j];a[j]=k;
            }
        }
    }
    for(i=0;i<N;i++){
        m=a[i];
        for(j=N-1;j>=0;j--){//倒序进行查找，m越大，M越大
            if(a[j]<=m*p){
                M=a[j];
                sum=j-i+1;
                break;
            }
        }
        if(sum>=max)
            max=sum;
        if(M==a[N-1])//如果M直接等于最后一个元素了，当前m到M就是完美数列中最大数字个数
            break;
    }
    cout<<max;
    return 0;
}//但是还是运行超时了，最后查找最大元素的算法时间复杂度还是太大

简便通过的代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main(){
    long int N,p,cnt=0;//cnt用于存放完美数列的数字个数
    cin>>N>>p;
    vector<int> v(N);//用vector容器声明了一个int数组，可以动态向里面添加删除
    for(int i=0;i<N;i++)
        cin>>v[i];
    sort(v.begin(),v.end());//直接用sort函数进行排序
    for(int i=0;i<N;i++){
        for(int j=i+cnt-1;j<N;j++){//让j每次都从i+cnt-1（就是上一次完美数列中最大的元素开始，因为m是从数列中最小的开始，越往后越大，那么完美数列中的最大的元素就越大），这样就会减小算法的时间复杂度，测试点4就可以通过
            if(v[j]>p*v[i]) break;
            if(j-i+1>cnt) cnt=j-i+1;
        }
    }
    cout<<cnt;
    return 0;
}

• 注意要使用c++中自带的函数简化代码