问题描述
小蓝对一个数的数位之和很感兴趣, 今天他要按照数位之和给数排序。当两个数各个数位之和不同时, 将数位和较小的排在前面, 当数位之和相等时, 将数值小的排在前面。
例如, 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 。
扫描二维码关注公众号,回复:
14589400 查看本文章
运行限制
- 最大运行时间:3s
- 最大运行内存: 512M
Python代码
n=int(input())
m=int(input())
s=list(range(1,n+1))
s.sort(key=lambda x: (sum(int(i) for i in str(x)),x)) # 核心代码
print(s[m-1])
这个代码只能通过90%的测试用例,对于10^6的数量级显然是有心无力。