PAT乙级1030：完美数列 (25）

题目

给定一个正整数数列，和正整数 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. 读入数据后对输入数据进行快速排序（我采用的是升序）;
2. 从后向前遍历，不断将遍历到的当前元素假设为最大值（外层循环），再以当前元素为下标上限，向前遍历（内层循环），找到第一个不满足题意的元素计算当前的“完美数列”长度，存入max；
3. 遍历结束，输出max。

难点

测试点5运行超时。

解决：将解题思路中所述的内层循环的起始点进行修改，具体来说：

由于只需要找到最长的完美数列，因此，内层循环的每一次遍历的时候都可以从当前元素（外层循环假设为最大值的元素，假设下标为i）向前数（max-1）个元素开始进行，如果该元素已经不满足条件，说明当a[ i ]为最大值时，该“完美数列”的长度已经不可能超越max了；如果该元素满足条件，那么继续向前遍历寻找是否存在更长的“完美数列”，也就是更大的max。这样对于每一个外层循环，都能够省下max次的循环。

原超时代码（需要修改部分）

for (j=i-1;j>=0;j--)
{
	if (a[j]<temp)
    {
    	k += (i-j-1);
    	break;
    }
}

修改为

for (j=i-max;j>=0;j--)//此处的j起始值进行修改
{
	if (a[j]<temp)
    {
    	k += (i-j-1);
    	break;
    }
}

代码

#include<stdio.h>
#include<stdlib.h>
int cmp(const void *a,const void *b){
    return *(int *)a - *(int *)b;//升序
}
int main(){
    int N,p,i,k,max,j;
    long double temp;
    scanf("%d %d\n",&N,&p);
    int a[N];
    for (i=0;i<N;i++)
        scanf("%d",&a[i]);
    qsort(a,N,sizeof(int),cmp);
    max = 0;
    if (p==0)
        printf("0\n");
    else
    {
        for (i=N-1;i>=0;i--)//从最后一个数字开始遍历，假设最后一个是最大值
        {
            k = 1;
            temp = a[i]*1.0/p;
            for (j=i-max;j>=0;j--)
            {
                if (a[j]<temp)
                {
                    k += (i-j-1);
                    break;
                }
            }
            if (j==-1)//如果是正常结束循环
                k+=(i-j-1);
            if (k>max)
                max = k;
            if (max>=(i+1))//如果i之前（包含i）的数字个数已经无法超过max了，那么就停止运行
                break;
        }
        printf("%d\n",max);
    }
    return 0;
}

二刷代码

#include<stdio.h>
#include<stdlib.h>
int cmp(const void *a, const void *b){
    return *(int *)a-*(int *)b;//升序ascending order
}
int main(){
    int i,j,p,N,len = 1;
    scanf("%d %d",&N,&p);
    int a[N];
    for (i=0;i<N;i++)
        scanf("%d",&a[i]);
    qsort(a,N,sizeof(a[0]),cmp);
    for (i=N-1;i>=0;i--)//倒叙，每一个遍历到的作为最大的数字
    {
        for (j=i-len;j>=0;j--)//往前找第一个不符合的数字
            if (a[j]<(a[i]*1.0/p))
                break;
        if ((i-j)>len)
            len = i-j;
    }
    printf("%d",len);
    return 0;
}