PAT(basic level) 1030 完美数列(25)

给定一个正整数数列，和正整数p，设这个数列中的最大值是M，最小值是m，如果M <= m * p，则称这个数列是完美数列。
现在给定参数p和一些正整数，请你从中选择尽可能多的数构成一个完美数列。

输入格式：

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

输出格式：

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

输入样例：

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
;
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;
}