给定整数 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
代码实现:
class Solution {
public:
int findKthNumber(int n, int k) {
int cur = 1;
--k;//初始化为cur = 1,k需要自减1
while (k > 0) {
long long step = 0, first = cur, last = cur + 1;
//统计这棵子树下所有节点数(step)
while (first <= n) {
step += min((long long)n + 1, last) - first;//不能超过n的值,并不是所有节点都有十个子节点
first *= 10;
last *= 10;
}
if (step <= k) {//不在子树中
++cur;
k -= step;
}
else {//在子树中,进入子树
cur *= 10;
--k;
}
}
return cur;
}
};