Python:每日一题之数位排序

问题描述

小蓝对一个数的数位之和很感兴趣, 今天他要按照数位之和给数排序。当 两个数各个数位之和不同时, 将数位和较小的排在前面, 当数位之和相等时, 将数值小的排在前面。

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

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

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

输入格式

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

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

输出格式

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

样例输入

13
5

样例输出

3

样例说明

1 到 13 的排序为: 1,10,2,11,3,12,4,13,5,6,7,8,9 。第 5 个数为 3 。

评测用例规模与约定

对于 30% 的评测用例, 1 ≤ m ≤ n ≤300 。

对于 50% 的评测用例, 1 ≤ m ≤ n ≤ 1000 。

对于所有评测用例, 1 ≤ m ≤ n ≤ 10 ^ 6 。

运行限制

• 最大运行时间：3s
• 最大运行内存: 512M

 参考代码：

n=int(input())
m=int(input())
a=list(range(1,n+1))
a.sort(key=lambda x:sum(int(i) for i in str(x)))
print(a[m-1])

---

用sort（）对每一项和进行快速排序