LeetCode —— 440. 字典序的第K小数字（Python）

给定整数 n 和 k，找到 1 到 n 中字典序第 k 小的数字。

示例 :
输入:
n: 13 k: 2

输出:
10

解释:
字典序的排列是 [1, 10, 11, 12, 13, 2, 3, 4, 5, 6, 7, 8, 9]，所以第二小的数字是 10。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/k-th-smallest-in-lexicographical-order
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
————————

考虑到时间消耗，这题使用深度遍历会超时，因此我们不能使用普通的递归方法，我们使用数学方法统计从一个节点遍历到另一个节点需要的步数。

class Solution:
    def findKthNumber(self, n: int, k: int) -> int:
        
        # cal_steps是用于统计从节点n1到节点n2所需要的步数
        def cal_steps(n, n1, n2):
        # n1和n2为当前层的两个不同节点
            step = 0
            while n1 <= n:
                step += min(n2, n + 1) - n1
                n1 *= 10
                n2 *= 10
            return step
                
        cur = 1  # 用于保存当前节点
        k -= 1  # 当k=0时返回当前cur节点
        
        while k > 0:
            steps = cal_steps(n, cur, cur + 1)  # 统计当前层从cur到cur+1需要的步数
            if steps <= k:  # 如果使用的步数小于等于最大值k，则统计当前层的cur+1到cur+2的步数，以此类推
                k -= steps
                cur += 1
            else:  # 如果使用的步数大于k，则进入下一层进行计算
                k -= 1
                cur *= 10
        
        return cur