1、题目描述
给定整数 n 和 k,找到 1 到 n 中字典序第 k 小的数字。
注意:1 ≤ k ≤ n ≤ 109。
2、示例
输入:
n: 13 k: 2
输出:
10
解释:
字典序的排列是 [1, 10, 11, 12, 13, 2, 3, 4, 5, 6, 7, 8, 9],所以第二小的数字是 10。
3、题解
首先我们可以通过简单的数学方式,计算出[n,n+1]之间有多少个数字。
初始化 l = 1, r = 9, k = k(还差数字个数)
令f(x)表示[x,x+1]之间数字个数
步骤1:遍历 i -> [l,r]
步骤2:
如果 k > i, k-=f(i).即第k个数不在[i,i+1]中,还需要找k-=f(i)个数。
如果 k <= i, k--,l=i10 ,r=i10+9,返回步骤1. 即第k个数在[i,i+1]中,还需找k-1个数,接下来查找区间为[i10,i10+9]。
class Solution {
public:
int findKthNumber(int n, int k) {
int l=1,r=9;
while(k)
{
for(int i=l;i<=r;i++)
{
int f=getNodeCnt(i,n);
if(k>f)
k-=f;
else
{
k--;
if(k==0) return i;
l=i*10+0;
r=i*10+9;
break;
}
}
}
return 0;
}
int getNodeCnt(int n,int maxn)
{
long left=(long)n*10+0,right=(long)n*10+9;
int res=1;
while(maxn>=left)
{
if(maxn<=right)
res+=maxn-left+1;
else
res+=right-left+1;
left=left*10+0;
right=right*10+9;
}
return res;
}
};