题目 2684: 蓝桥杯2022年第十三届省赛真题-数位排序 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 ≤ 106。 

源代码

n = int(input())
m = int(input())
a = []
for i in range(1, n+1):
    if i < 10:
        a.append((i, i))
    else:
        sum1 = 0
        c = i
        while c:
            sum1 += c % 10
            c //= 10
        a.append((sum1, i))
        # a.append((sum(list(map(int, str(i)))), i))
# print(a)
b = sorted(a)
print(b[m-1][1])