蓝桥杯-数位之和

问题描述

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

例如, 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,91,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≤10e6 。

运行限制

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

因为案例最大是10的6次，但实际上因为是数位之和，所以最大值最多就是6X9=54，所以我是直接建了个二维数组，行数就是数位之和，遍历数据，当一个数数位之和达到行的下标 （因为第一行是li[0]），就append 。由题意，数位之和相同的数，肯定是小的在前，大的在后，数位和不同的数从小到大依次往下排。有个问题就是，10e6超时了，最后一个用例通不过，10e5级别都是可以的。

n = int(input())
m = int(input())
n= n+1
li = [[] for i in range(54)]

for i in range(n):
    sum = 0
    s = str(i)
    for j in range(len(s)):
        sum += int(s[j])
    li[sum].append(i)
x = 0
for i in range(1,len(li)):
    if(m>len(li[i])):
        m -= len(li[i])
        # print(m)
        x+=1
    else:break
# print(li)
# print(x,m)
if m ==0 :
    print(li[x-1][len(li[x-1])-1])
else:
    print(li[x+1][m-1])

别人的题解

def key(a):
    ans = 0
    for k in str(a):
        x = int(k)
        ans += x
    return ans
n = int(input())
m = int(input())
a = []
for i in range(1, n + 1):
    a.append(i)
a = sorted(a, key=lambda x: key(x))
print(a[m - 1])