PTA 完美数列（c++）

1030 完美数列 (25分)

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

输入格式：

输入第一行给出两个正整数 N 和 p，其中 N（≤10​5​​）是输入的正整数的个数，p（≤10​9​​）是给定的参数。第二行给出 N 个正整数，每个数不超过 10​9​​。

输出格式：

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

输入样例：

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

输出样例：

8

样例解答：

不出所料java又超时了呀，查阅一番后没有找到解答，就用c++写啦。

#include <bits/stdc++.h>
using namespace std;

int main() {
 int n;
 long long p; 
 cin>>n>>p;
 int a[n];
 for(int i=0;i<n;i++) {
  cin>>a[i];
 }
 //先排序，方便操作
 sort(a,a+n);
 int max=0;
 for(int i=0;i<n;i++) {
 //防止点4超时，这里直接给j加上目前求出的最大值max
 //如果i+max已经大于n可以直接结束了
  if(i+max>n) break;
  for(int j=i+max;j<n;j++){
   if(a[j]>a[i]*p){
    break;
   }
   if(j-i+1>=max){
    max=j-i+1;
   }
  }
 }
 cout<<max<<endl;
 return 0;
}