440.字典序的第K小数字

题目

440.字典序的第K小数字

题目大意

给定整数 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

数据规模

提示:

• $1<=k<=n<=10^9$

思路

利用字典树的特性将所有小于等于$n$的数字按照字典序的方式进行重建，可以得到如下：（借用leetcode的官方题解的图）

很显然，通过前序遍历就可以得到字典序从小到大的序列。遍历到第$i$个节点就是第$i$小的数字。对于节点$i$，它的子节点的数字为$(10\times i,10\times i+1,...,10\times i+9)$。设当前的字典树的第$i$小的节点值为$now$，只需按照前序遍历再继续往后找$k-i$个节点即为目标节点。设与$now$右侧第一个兄弟节点为$now+1$，前序遍历先遍历$now$的子树，再遍历$now+1$的子树。设$now$为根节点的子树的节点数为$cnt=Count(now,n)$：

• 如果$cnt<=k$，则此时可以肯定第$k$小的节点一定不在以$now$为根的子树中，而在$now$的右侧兄弟节点的子树中，则可以跳过$cnt$个节点。从$now+1$继续查询。
• 如果$cnt>k$，则此时可以肯定第$k$小的节点一定在以$now$为根的子树中。从$now$儿子节点的左侧第一个孩子开始往后查找$k-i-1$个节点，此时左侧第一个孩子的值为 $10\times now$。
• 依次重复上述操作直到找到$k$小的节点即可。

ll Count(ll cur,ll n)函数：按照层次遍历子树，$s$指向第$ i $层的最左侧的孩子节点， $t$指向第 $i$ 层的最右侧的孩子节点，那么$s=s\ast 10,t=t\ast 10+9$，第$i$层共$t-s+1$个节点，但是注意$t=min(t,n)$。不断循环直到$s>n$终止循环。

代码

// short int long float double bool char string void
// array vector stack queue auto const operator
// class public private static friend extern 
// sizeof new delete return cout cin memset malloc
// relloc size length memset malloc relloc size length
// for while if else switch case continue break system
// endl reverse sort swap substr begin end iterator
// namespace include define NULL nullptr exit equals 
// index col row arr err left right ans res vec que sta
// state flag ch str max min default charray std
// maxn minn INT_MAX INT_MIN push_back insert
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int>PII;
typedef pair<int, string>PIS;
const int maxn=1e5+50;//注意修改大小
long long read(){long long x=0,f=1;char c=getchar();while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}while(isdigit(c)){x=x*10+c-'0';c=getchar();}return x*f;}
ll qpow(ll x,ll q,ll Mod){ll ans=1;while(q){if(q&1)ans=ans*x%Mod;q>>=1;x=(x*x)%Mod;}return ans%Mod;}

class Solution {
public:
    ll Count(ll cur,ll n){
        ll cnt=0,s=cur,t=cur;
        while(s<=n){
            cnt+=min(t,n)-s+1;
            s*=10;
            t=t*10+9;
        }
        return cnt;
    }
    int findKthNumber(int n, int k) {
        int now=1;
        k--;
        while(k){
            int cnt=Count(now,n);
            if(cnt<=k){
                k-=cnt;
                now++;
            }
            else{
                now*=10;
                k--;
            }
        }
        return now;
    }
};