zyb的面试 字节跳动-文远知行杯”广东工业大学第十四届程序设计竞赛

Problem Description

今天zyb参加一场面试,面试官听说zyb是ACMer之后立马抛出了一道算法题给zyb:
有一个序列,是1到n的一种排列,排列的顺序是字典序小的在前,那么第k个数字是什么?
例如n=15,k=7, 排列顺序为1, 10, 11, 12, 13, 14, 15, 2, 3, 4, 5, 6, 7, 8, 9;那么第7个数字就是15.
那么,如果你处在zyb的场景下,你能解决这个问题吗?

Input

T组样例(T<=100)
两个整数n和k(1<=n<=1e6,1<=k<=n),n和k代表的含义如上文

Output

输出1-n之中字典序第k小的数字

Sample Input
1
15 7
 

Sample Output
15

思路：以前做过一道类似的题，当初补题时候发现了一个大佬写的模板，跟这道题问的是一样的，收藏了起来，直接一套就A了

附上大佬链接：https://blog.csdn.net/hanbingsqq/article/details/79678964

代码如下：

#include
#define ll long long
using namespace std;
ll getsteps(ll n,ll a,ll b)
{
    ll res=0;
    while(a<=n)
    {
        res=res+min(n+1,b)-a;
        a=a*10;
        b=b*10;
    }
    return res;
}
ll getans(ll n,ll cur,ll steps)
{
    while(steps>0)
    {
        ll curstep=getsteps(n,cur,cur+1);
        if(curstep<=steps)
        {
            steps=steps-curstep;
            cur++;
        }
        else
        {
            steps--;
            cur*=10;
        }
    }
    return cur;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        printf("%d\n",getans(n,1,k-1));
    }
    return 0;
}