1030. 完美数列(25) PAT

1030. 完美数列(25)

时间限制

300 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CAO, Peng

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

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

8

#include <cstdio>
#include <algorithm>
#define MAX 100005
using namespace std;
//大数据样例没有通过 
int main(){
	long long a[MAX],p;
	int N;
	scanf("%d%lld",&N,&p);
	for(int i=0;i<N;i++){
		scanf("%lld",&a[i]);
	}
	sort(a,a+N);
	int num=0,maxn=0;
	long long flag;
	for(int i=0;i<N;i++){//这里的两个嵌套的for循环时间复杂度太高，需要优化
		flag=a[i]*p;
		if(maxn>=(N-i)){
			break;
		}
		num=0;
		for(int j=i;j<N;j++){
			if(a[j]<=flag){
				num++;
			}else{
				break;
			}
		}	
		if(maxn<num){
			maxn=num;
		}
	}
	printf("%d",maxn);
	return 0;
} 

#include <cstdio>
#include <algorithm>
#define MAX 100005
using namespace std;
//AC代码
int main(){
	long long a[MAX],p;
	int N;
	scanf("%d%lld",&N,&p);
	for(int i=0;i<N;i++){
		scanf("%lld",&a[i]);
	}
	sort(a,a+N);
	int num=0,maxn=0;
	long long flag;
	for(int i=0;i<N;i++){
		flag=a[i]*p;
		for(int j=(i+maxn-1);j<N;j++){//和上个代码不一样，在这里优化了内部for循环，直接从最长的可能开始找起，而不是像上面，每一次都挨个挨个循环
			if(a[j]>flag){
				break;
			}else{
				if(maxn<(j-i+1))
				maxn=j-i+1;
			}
		} 
	}
	printf("%d",maxn);
	return 0;
}