2022第十三届蓝桥杯省赛B组Python 试题 D: 数位排序

题目

小蓝对一个数的数位之和很感兴趣，今天他要按照数位之和给数排序。当 两个数各个数位之和不同时，将数位和较小的排在前面，当数位之和相等时， 将数值小的排在前面。 例如，2022 排在 409 前面，因为 2022 的数位之和是 6，小于 409 的数位 之和 13。 又如，6 排在 2022 前面，因为它们的数位之和相同，而 6 小于 2022。 给定正整数 n，m，请问对 1 到 n 采用这种方法排序时，排在第 m 个的元 素是多少？

代码：

from functools import cmp_to_key
n=int(input())
m=int(input())

ltmp=[i for i in range(1,n+1)]
def sorts(x,y):
    x1=sum(list(map(int,list(str(x)))))
    y2=sum(list(map(int,list(str(y)))))
    if x1>y2:
        return -1
    if x1==y2:
        if x>y:
            return -1
    else:
        return 0
ltmp.sort(key=cmp_to_key(sorts))
ltmp=ltmp[::-1]
print(ltmp[m-1])

用到的是自定义排序，应该是没错的.