小米OJ 128.不一样的排序

题目描述：

有一天利姆鲁想教他的哥布林部下学数学，因为他之前教过因子，现在想考考他们。

利姆鲁问现在有 n 个数，需要用因子个数的多少进行排序，因子个数多的排在后面，因子个数少的排在前面，如果因子个数相同那么就比较这个数的大小，数大的放在后面，数小的放在前面。现在让你说出排序之后第 K 个位置的数字是多少。

输入描述：

第 1 个整数为整数 $K$，$1\le K\le 10^6$；

第 2 个为整数 $n$，表示数字的数量，$n<10^7$；

接下来有$n$个整数，每个数的大小不超过$10^6$.

输出描述：

输出排序之后的第 K 位置的数值。

输入样例：

4 6 1 2 3 4 5 6

输出样例：

5

思路

因子个数打表后排序即可。

更多题解请关注 doubleQ2018’s GitHub

ac代码

#include <bits/stdc++.h>

typedef long long ll;
using namespace std;

#define maxn 1000000
int cc[maxn+5] = {0};
int nums[10000005] = {0};
void init(int n)
{
    int x = sqrt(n);
    for (int i = 1; i <= x; i++){
        for (int j = i + 1; j * i <= n; j++) cc[i * j] += 2;
        cc[i * i]++;
    }
}

int main(){
    init(maxn);
    int k, n;
    cin >> k >> n;
    for(int i = 0; i < n; i++) cin >> nums[i];
    sort(nums, nums+n, [](const int &x, const int &y){
            if(cc[x] != cc[y]) return cc[x] < cc[y]; else return x < y;});
    cout << nums[k-1] << endl;
}

更多题解请关注 doubleQ2018’s GitHub