蓝桥杯：数位排序（C++）

禁止转载，该文章内容并非教学，仅为个人笔记。

问题描述

小蓝对一个数的数位之和很感兴趣，今天他要按照数位之和给数排序。当两个数各个数位之和不同时，将数位和较小的排在前面，当数位之和相等时，将数值小的排在前面。
例如，2022 排在 409 前面，因为 2022 的数位之和是 6，小于 409 的数位之和 13。
又如，6 排在 2022 前面，因为它们的数位之和相同，而 6 小于 2022。
给定正整数 n，m，请问对 1 到 n 采用这种方法排序时，排在第 m 个的元素是多少？
【输入格式】
输入第一行包含一个正整数 n。
第二行包含一个正整数 m。
【输出格式】
输出一行包含一个整数，表示答案。
【样例输入】
13
5
【样例输出】
3
【样例说明】
1 到 13 的排序为：1, 10, 2, 11, 3, 12, 4, 13, 5, 6, 7, 8, 9。第 5 个数为 3。
【评测用例规模与约定】
对于 30% 的评测用例，1 ≤ m ≤ n ≤ 300。
对于 50% 的评测用例，1 ≤ m ≤ n ≤ 1000。
对于所有评测用例，1 ≤ m ≤ n ≤ 10^6。

解题思路

1. 重写 algorithm 头文件中的 sort 方法的排序规则，对 vector 容器进行排序即可。
2. 用 map 容器，以数值为 key，满足数值的数放进 vector 作为其 value，计算每个数即可。

也许第二种方法会更快一点，也不一定

代码实现

方法一 排序

#include<iostream>
#include <vector>
#include <algorithm>
using namespace std;
int n, m;
vector<int> s;
int fun(int a)
{
    int res = 0;
    while (a)
    {
        res += a % 10;
        a /= 10;
    }
    return res;
}
int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; ++i)
        s.push_back(i);
    sort(s.begin(), s.end(), [](int a, int b){ return fun(a) < fun(b); });
    cout << s[m - 1];
    return 0;
}

方法二 遍历

#include<iostream>
#include <map>
#include <vector>
using namespace std;
int n, m;
map<int, vector<int>> Dic;
int fun(int a)
{
    int res = 0;
    while (a)
    {
        res += a % 10;
        a /= 10;
    }
    return res;
}

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; ++i)
    {
        int t = fun(i);
        if (Dic.find(t) == Dic.end())
            Dic.insert(pair<int, vector<int>>(t, vector<int>(0)));
        Dic.at(t).push_back(i);
    }

    for (const auto &i: Dic)
    {
        if (m > i.second.size())
            m -= i.second.size();
        else
        {
            cout << i.second.at(m - 1);
            break;
        }
    }
    return 0;
}