C++ - PAT - 1030. 完美数列(25)

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

看完题目最容易想到的是：先将数列排列后， for i循环中嵌套一个for  j找出能选出的最大数  。

1.N的范围是(<=10^9) N^2会达到10^18，可能会超时。用ans记录当前最大能选择的个数，当进行下一次 i 判断的时候，就可以减少不必要的判断，从而达到优化的效果。

2.正整数是不超过10^9，p也是，p*num最大会达到10^18,如果int的范围是10^16那么就会超范围溢出。所以要换用long或者double都可以。

#include<cstdio> 
#include<algorithm>
using namespace std;

int main(){
	int ans=1; 
	long n,p,temp;
	long M ,m;	
	//读取和存储数据 
	scanf("%ld%ld",&n,&p);
	
	int num[n];
	
	for(int i=0 ;i<n ;i++){
		scanf("%ld",&num[i]);
	}
	
	sort(num,num+n);
	//遍历 
	for(int i=0 ;i<n ;i++){
		m = num[i];
		for(int j=i+ans ;j<n ;j++){ //优化 
			M = num[j];
			if(M<=p*m){
				if(j-i+1>ans){
					ans = j-i+1;
				}
			}else{
				break;
			}
		}
	}
	
	printf("%d\n",ans);
	
	return 0;
}