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

思路：排序后，穷举所有的排列，判断最大值和最小值的关系是否符合完美数列。

优化：用ans记录能选择的最多的数的个数。下次判断直接+ans  ，这样可以减少时间开销，具体看代码中的实现。

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

int main(){
  	int ans=1; 

  	long n,p,temp;
  	long M ,m;  //两数相乘会超出int范围 
  	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;
}