Acwing 4653. 数位排序

题目链接：4653. 数位排序 - AcWing题库

思路：写一个自定义排序即可

代码：

#include<iostream>
#include<algorithm>
using namespace std;

const int N = 1000005;
struct Num
{
    int num;
    int sum;
    bool operator<(const Num a)const{
        if(sum!=a.sum) return sum<a.sum;
        return num<a.num;
    }
}a[N];

int n,m;
int tmp;

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;++i)
    {
        a[i].num=i;
        tmp=i;
        while(tmp)
        {
            a[i].sum+=tmp%10;
            tmp/=10;
        }
    }
    
    sort(a+1,a+n+1);
    
    cout<<a[m].num<<endl;
    
    return 0;
}