PAT(basic level) 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

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

int main()
{
	int n, p;
	cin >> n >> p;
	int a[n];
	for(int i = 0; i < n; i ++)  cin >> a[i];
	
	sort(a, a+n);   //排序 
	long long M;
	int count = n;
	for(int i = n-1; i >= 0; i --){  //找以第一个数为最小值的最大长度 
		M = a[0]*p;
		if(a[i] > M) count --;
		else break;
	}

	for(int i = 1; i < n-count+1; i ++){    //查看以中间的数为最小值的长度       
        M = a[i]*p;									 
		for(int j = i+count; j < n; j ++){   //直接看最大长度后的数，提高效率 
			if(a[j] <= M) count ++;
			else break;
		}
	}                                         
	cout << count;
	return 0;
}