[leetcode] 1030 完美數列

題目看起來不是很複雜，先排序後，再找到一個儘可能長的數列：

給定一個正整數數列，和正整數 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

爲了減少查詢量，j = i + ans，使得每次查詢的都是更長的滿足要求的。

代碼爲：

#include<cstdio>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;

long long a[100001];

int main()
{
    long long N, p;
    scanf("%lld%lld", &N, &p);
    for(int i = 0; i < N; i++)
        scanf("%lld", &a[i]);
    long long ans = 0;
    sort(a, a + N);
    for(int i = 0; i < N; i++)
    {
        for(int j = i + ans; j < N; j++)
        {
            if(a[j] > a[i] * p)
                break;
            if(ans < j - i + 1)
                ans = j - i + 1;
        }
    }
    printf("%lld\n", ans);
    return 0;
}