题解:P12366 [蓝桥杯 2023 省 Python B] 数位排序

· · 题解

P12366 [蓝桥杯 2023 省 Python B] 数位排序

解题思路

将数字与数位和捆绑(使用结构体或元组)放入数组或列表,以数位和为关键字排序,最后输出第 m 个数。

完整代码

C++:

#include<stdio.h>
#include<algorithm>
using namespace std;
const int MAXN=1e6+7;
struct Node{
    int id,num;
}l[MAXN];
int n,m,ans;
inline int Count(int x)
{
    int sum=0;
    while(x)sum+=x%10,x/=10;
    return sum;
}
inline bool cmp(Node A,Node B)
{
    return A.num==B.num?A.id<B.id:A.num<B.num;
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)l[i]={i,Count(i)};
    sort(l+1,l+1+n,cmp);
    printf("%d",l[m].id);
    return 0;
}

Python:

def takeSecond(elem):
    return elem[1]
def count(x):
    tmp,sum=x,0
    while tmp>0:
        sum+=tmp%10
        tmp//=10
    return sum
l=[]
n,m=int(input()),int(input())
for i in range(n):
    l.append((i+1,count(i+1)))
l.sort(key=takeSecond)
print(l[m-1][0])