蓝桥杯 数位排序 用例全过版 Python

前言：

这题要想测试用例全部通过时间复杂度需要达到 O(N) 否则最后几个测试用例会超时

解题思路：

首先得了解一些基本概念

1. 数位和：

数位和指的是一个数字，将他各个位数字相加的和。 例如：123 的数位和是 6， 因为1 + 2 + 3 = 6

这题我使用哈希表来解决的，题目说“数位和小的排前面，数值小的排前面”，那么我们直接将 key 当作数位和， values 里面存储每个数值。

最后遍历values找到 m 就可以了

代码

# defaultdict ->    key: list
from collections import defaultdict

# input
n = eval(input())
m = eval(input())


# function
def f(num: int) -> int:
    SUM = 0
    while num:
        SUM += num % 10
        num //= 10
    return SUM


dic = defaultdict(list)
for num in range(1, n + 1):
    SUM = f(num)
    # number sum
    dic[SUM].append(num)

# search m
before, now = 0, 0
for lis in dic.values():
    now = len(lis)
    if before + now >= m:
        print(lis[m - before - 1])
        break
    before += now