题目: 给定整数 n 和 k,找到 1 到 n 中字典序第 k 小的数字。
注意:1 ≤ k ≤ n ≤ 109。
示例 :
输入:
n: 13 k: 2
输出:
10
解释:
字典序的排列是 [1, 10, 11, 12, 13, 2, 3, 4, 5, 6, 7, 8, 9],所以第二小的数字是 10。
这道题目感觉有点难
(1)首先列出所有的字典序,然后选择第k个
class Solution {
public:
bool static compare(int a,int b)
{
return std::to_string(a) < std::to_string(b);
}
int findKthNumber(int n, int k) {
vector<int> Pool(n, 0);
for (int i=0;i<n;i++)
{
Pool[i] = i+1;
}
sort(Pool.begin(),Pool.end(),compare);
return Pool[k-1];
}
};
可以看到,非常的简介明了,主要是用了sort()进行了排序,排序的算法就是转换为字符串,然后比较非常的简单明了。可惜了,这个的时间复杂度过不去,耗时要比较久。但是,这个给了我们生成字典序的一种方式。可以通过string比较的方式,快速生成字典序。
(2) 用树的方式
如下图,所示,整个的字典序的排序方式,就是这个10叉树的一个优序遍历。每一个节点都拥有 10 个孩子节点,因为作为一个前缀 ,它后面可以接 0~9 这十个数字。而且你可以非常容易地发现,整个字典序排列也就是对十叉树进行先序遍历。1, 10, 100, 101, ... 11, 110
所以,如果我们可以计算出每个前缀下面有多少个数字,那么就课可以很快的完成这个事情,比如说 1的前缀下面的有 1 10 11 12 ....18 19 100 101 102 ......198 199 等。我们是可以计算出来。
getTheNumberOfFix(prefix,n) 就是计算当前的前缀下,有多少个数字,prefix就是前缀,n是上限。主要的思路是当前的前缀的和下一个的前缀的之间的数量和,就是当前前缀的所有的个数。
class Solution {
public:
int getTheNumberOfFix(int prefix, int n)
{
long long currentFix=prefix;
long long nextFix=currentFix+1;
int totolNumber=0;
while(currentFix<=n)
{
totolNumber +=nextFix-currentFix;
currentFix *=10;
if(nextFix*10<n+1)
{
nextFix=nextFix*10;
}
else
{
nextFix=n+1;
}
}
return totolNumber;
}
int findKthNumber(int n, int k) {
int currentPosition=1;
long long prefix=1;
while(currentPosition<k)
{
int numberOFcurrentFix=getTheNumberOfFix(prefix,n); // 获取当前的前缀下的所有的数字的个数
if(numberOFcurrentFix+currentPosition>k) // 说明 k 在当前的前缀下
{
prefix *=10;
currentPosition++;
}else // k 在下一个前缀
{
prefix++;
currentPosition +=numberOFcurrentFix;
}
}
return prefix;
}
};
链接:https://leetcode-cn.com/problems/k-th-smallest-in-lexicographical-order/solution/ben-ti-shi-shang-zui-wan-zheng-ju-ti-de-shou-mo-sh/