PAT.B1030 完美数列

返回目录

题意

从N个正整数中选择若干个数，使得选出的这些数中的最大值不超过最小值的p倍。问满足条件的选择方案中，选出的数的最大个数。

样例（可复制）

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

样例输出

8

提供易错样例

3 3
1 2 3
//output 3

注意点

1. 由于题干中涉及序列的最大值和最小值，因此不妨先将所有N个正整数从小到大进行排序。在此基础上证明:能使选出的数个数最大的方案，一定是在该递增序列中选择连续的若干个数的方案。
2. p与序列中的元素均可能达到10**9，因此a[i] *p可能达到10的18次方， 所以必须使用long long类型
3. 如果强制进行O(n**2)的二重循环枚举，那么根据题目的数据范围，肯定是会超时的。以下用两种方法来解决这个问题:二分查找和two pointers。
4. 二分法里用了upper_bound(),该函数底层是使用二分法实现的，具体到本题会在a[i+1]~a[n-1]中查找第一个超过a[i]*p的数，返回其位置指针。也可以自己写个二分查找函数。
5. 根据实际测试，两者的时间复杂度基本为线性，相比之下，two pointers略快一点

二分法

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

int a[100010];

int main(){
    int count=1,n,p;
    scanf("%d%d",&n,&p);
    for(int i=0;i<n;i++)scanf("%d",&a[i]);
    sort(a,a+n);
    for(int i=0;i<n;i++){
        int j=upper_bound(a+i+1,a+n,(long long)a[i]*p)-a;
        count=max(count,j-i);
    }   
    printf("%d",count);
    return 0;
}

two pointers

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

int a[100010];

int main(){
    int count=0,n,p;
    scanf("%d%d",&n,&p);
    for(int i=0;i<n;i++)scanf("%d",&a[i]);
    sort(a,a+n);
    int i,j=0;
    for(i=0;i<n;i++){
        while(j<n&&(long long)a[j]<=(long long)a[i]*p){
            j++;
        }
        count=max(count,j-i);
    }   
    printf("%d",count);
    return 0;
}