今日头条——校招在线编程题《字典序》

题目描述
给定整数n和m, 将1到n的这n个整数按字典序排列之后, 求其中的第m个数。
对于n=11, m=4, 按字典序排列依次为1, 10, 11, 2, 3, 4, 5, 6, 7, 8, 9, 因此第4个数是2.

输入描述:
输入仅包含两个整数n和m。

数据范围:

对于20%的数据, 1 <= m <= n <= 5 ;

对于80%的数据, 1 <= m <= n <= 10^7 ;

对于100%的数据, 1 <= m <= n <= 10^18.
输出描述:
输出仅包括一行, 即所求排列中的第m个数字.

C++代码：

//主要采用字典树的思想
#include<iostream> 
using namespace std;

long get_num(long long k,long long n){
    long long base=1,count=0;
    while((k+1)*base-1<=n){
        count+=base;
        base*=10;
    }
    if(k*base<=n)
        count+=n-k*base+1;
    return count;
}
long long ans(long long n,long long m){
    long long k=1;
    while(m){
        long long count=get_num(k,n);
        if(count>=m){
            m--;
            if(!m)  break;
            k*=10;
        } 
        else{
            k++;
            m-=count;
        }   
    }
    return k;
}
int main(){
    long long n,m;
    cin>>n>>m;
    cout<<ans(n,m)<<endl;
    return 0;
}