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
看到字典序查询,似乎第一印象是字典树。而这里正好需要它的改版——0~9字典树(临时起名)。而且不同于普通tire tree(字典树),这个题里只要假设它存在就好了。什么叫假设存在呢?就是我们用最大数值来设定一个边界限制深度搜索。拿样例为例,我们从1开始进位搜索(因为是dfs),搜到11,再搜到21,这个时候大于15,直接退出(return ;),每一次成功的深搜(不退出)都要对步数减一,步数为0的时候就找到了所需的数字。
那么显然地,有这样的思路:
- 对已有数据范围内进行模拟十叉树式搜索,每一次成功的搜索都要对步数减一,否则直接退出函数。特别注意一点,就是不要把步数参数放在dfs函数的参数列表里。因为在递归的时候只有深入一层