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

注意：

• 注意資料範圍，定義為long long,防止相乘資料溢位

程式碼：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int cmp(int a, int b){return a < b;}
int getMax(int a, int b){return a>b ? a : b;}
int main()
{
    int n;
    long long p;
    cin >> n >> p;
    if(n == 0){
        cout << n;
        return
 
 0;
    }
    vector<long long int> arr(n);
    for(int i=0; i<n; i++)
        cin >> arr[i];
    sort(arr.begin(), arr.end(), cmp);
    int res=1;
    for(int i=0; i<n-1; i++){ //遍歷，將arr[i]作為最小數
        for(int j=i+res; j<n; j++){//j置為要滿足可以更新數列長度的值，減少迴圈次數
            if(arr[i]*p < arr[j])//如果不滿足條件了，則將下一個元素arr[i+1]最為最小值
 

                break;
            res = getMax(res, j-i+1);//如果此次的長度大於上一次，更新數列長度
        }
    }
    cout << res << endl;
    return 0;
}