440字典序的第K小数字（字典树）

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;
    }
};