【PAT乙级】1030 完美数列

题目

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

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

8

思路

1. 首先使用< alogrithm >的sort将数组排序。
2. 使用10⁹/p计算溢出上限值a。
3. 从数组首位开始，使用分而治之递归的思路查找符合小于M条件的最大值位置，统计完美数列个数。
4. 查询位达到溢出上限后，可以停止后续查找，因为数列个数只会越来越少。

代码

#include<iostream>
#include<algorithm>
using namespace std;
#define X 1000000000

int imax(int num[],int M, int m, int n){//查找m-n范围内符合条件的最大值位置
    int x,mid = (m+n)/2;
    if(mid == m) x = num[n]>M?m:n;
    else if(num[mid] <= M) x = imax(num,M,mid,n);
    else x = imax(num,M,m,mid);
    return x;
}

int main(){
    int N,p,max=0;
    cin >> N >> p;
    int num[N];
    for(int i=0;i<N;i++) cin >> num[i];
    sort(num,num+N);
    int a = X / p;
    for(int i=0;i<N;i++){ //当以num[i]为m时
        int count = 0;
        if(num[i]>a){//若mp溢出，后续必小于此次循环，可停止判断
            count = N - i;
            if(count>max) max = count;
            break;
        }
        int M = num[i] * p;
        count = imax(num,M,i,N-1)-i+1;
        if(count>max) max = count;
    }
    cout << max << endl;
    
    return 0;
}

输出结果