【蓝桥杯】第十三届省赛-数位排序 （结构体排序 | sort函数的cmp）

题目描述

小蓝对一个数的数位之和很感兴趣，今天他要按照数位之和给数排序。

当两个数各个数位之和不同时，将数位和较小的排在前面，当数位之和相等时，将数值小的排在前面。

例如，2022 排在 409 前面，因为 2022 的数位之和是 6，小于 409 的数位之和 13。

又如，6 排在 2022 前面，因为它们的数位之和相同，而 6小于 2022。

给定正整数 n，m，请问对 1 到 n 采用这种方法排序时，排在第 m 个的元素是多少？

输入格式

输入第一行包含一个正整数 n。

第二行包含一个正整数 m。

输出格式

输出一行包含一个整数，表示答案。

数据范围

对于 30% 的评测用例，$1\le m\le n\le 300$。
对于 50% 的评测用例，$1\le m\le n\le 1000$。
对于所有评测用例，$1\le m\le n\le 10^6$。

输入样例：

13
5

输出样例：

3

样例解释

1 到 13 的排序为：$1,10,2,11,3,12,4,13,5,6,7,8,91,10,2,11,3,12,4,13,5,6,7,8,9$。

第 5 个数为 3。

思路

排序问题。考虑到要按照不同属性排序，先用结构体存储，对结构体进行自定义排序。

sort函数的cmp。

用结构体数组存储信息，a[i].x存原数，a[i].y存该数的各位数之和。

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1e6+10;
struct s
{
    int x, y;
}a[N];

bool cmp(s a, s b)
{
    if(a.y != b.y) return a.y < b.y;
    return a.x < b.x;
}
int main()
{
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        a[i].x = i;
        for(int t = i; t; t /= 10) a[i].y += t % 10;
    }
    sort(a+1, a+1+n, cmp);
    cout << a[m].x;
    return 0;
}