发布时间:2022-08-19 11:35
小蓝对一个数的数位之和很感兴趣,今天他要按照数位之和给数排序。当 两个数各个数位之和不同时,将数位和较小的排在前面,当数位之和相等时, 将数值小的排在前面。 例如,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])
用到的是自定义排序,应该是没错的.