22-03-23 字典序的第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

提示:

• 1 <= k <= n <= 10^9

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

---

解法一：计数模拟

今天的题目本来还想用基础的模拟看看，但是看了以下题目给出的范围，肯定会爆掉，只好参考一下大佬的解法了

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();
        int ans = 0, u = Integer.parseInt(b.substring(0,n)), k = m - 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;
    }
}

参考

---

此文章创于本人学习时的记录，如有错误或更优解还请指出