[leetcode]K-th Smallest in Lexicographical Order

最新推荐文章于 2026-05-25 10:17:00 发布

原创最新推荐文章于 2026-05-25 10:17:00 发布 · 511 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

当前文章被以下社区和专栏收录：

本文介绍了一种高效算法，用于找到1到n范围内字典序第k小的数字。通过逐步缩小搜索范围，该算法能够在较短时间内找到目标数字。

Description

Problem Link

给定n和k，求1..n中，字典序第k小的数。
例如：

Input:
n: 13   k: 2

Output:
10

Explanation:
The lexicographical order is [1, 10, 11, 12, 13, 2, 3, 4, 5, 6, 7, 8, 9], so the second smallest number is 10.

Solution

刚开始，令ans = 1，k–，然后每次计算ans，和ans+1之间有多少个数字。假设ans与ans + 1之间的数字个数gap，若gap <= k：则令ans = ans + 1，k = k - gap。否则令ans = ans * 10，k–；

附代码：

class Solution {
public:
   int getGap(int n,int p, int q) {
    int tmp = 0;
    while (p <= n) {
        tmp += max(0, min(n+1, q) - p);
        p *= 10;
        q *= 10;
    }
    return tmp;
}
int findKthNumber(int n, int k) {
    if (n == 0) return 0;
    int ans = 1;
    for (k -= 1; k > 0; ){
        int gap = getGap(n, ans, ans+1);
        if (gap <= k) {
            ans++;
            k -= gap;
        } else {
            ans *= 10;
            k -= 1;
        }
    }
    return ans;
}
};