PAT_乙级_1030

1030 完美数列

给定一个正整数数列，和正整数 p，设这个数列中的最大值是 M，最小值是 m，如果 M≤mp，则称这个数列是完美数列。

现在给定参数 p 和一些正整数，请你从中选择尽可能多的数构成一个完美数列。

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

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

输入样例：

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

输出样例：

8

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
bool cmp(long int a, long int b)
{
    return a > b;
}
vector<long int> nums;
int re[100005];
int main()
{
    long int num, p, temp;
    cin >> num >> p;
    int index = 0;
    for (int i = 0; i < num; i++) {
        scanf("%d", &temp);
        nums.push_back(temp);
    }
    for (int i = 0; i < 100005; i++) {
        re[i] = 0;
    }
    int sum_temp;
    sort(nums.begin(), nums.end());
    //核心
    for (int i = 0; i < num; i++) {
        for (int j = index; j < num; j++) {
            if (nums[j] <= p * nums[i]) {
                sum_temp = j - i + 1;
                re[i] = sum_temp;
                index = j;
            } else {
                break;
            }
        }
    }
    sort(re, re + num);
    cout << re[num - 1] << endl;
    return 0;
}