LeetCode刷题题解：440. 字典序的第K小数字

440. 字典序的第K小数字

题目：给定整数 n 和 k，返回 [1, n] 中字典序第 k 小的数字。

示例1：

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

示例2：

输入: n = 1, k = 1
输出: 1

解题思路：

计数模拟

Java代码：

class Solution {
    public int findKthNumber(int n, int k) {
        int ans = 1;
        while (k > 1) {
            int cnt = getCnt(ans, n);
            if (cnt < k) {
                k -= cnt; ans++;
            } else {
                k--; ans *= 10;
            }
        }
        return ans;
    }
    int getCnt(int x, int limit) {
        String a = String.valueOf(x), b = String.valueOf(limit);
        int n = a.length(), m = b.length(), k = m - n;
        int ans = 0, u = Integer.parseInt(b.substring(0, n));
        for (int i = 0; i < k; i++) ans += Math.pow(10, i);
        if (u > x) ans += Math.pow(10, k);
        else if (u == x) ans += limit - x * Math.pow(10, k) + 1;
        return ans;
    }
}

复杂度分析：

• 时间复杂度： 枚举前缀以及 getCnt 操作均与位数相关，复杂度均为 O(logn)。整体复杂度为O(log^2n)
• 空间复杂度： 忽略子串生成复杂度为 O(1)O(1)，否则为 O(log^2n)