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的场景下,你能解决这个问题吗?
T组样例(T<=100) 两个整数n和k(1<=n<=1e6,1<=k<=n),n和k代表的含义如上文
Output
输出1-n之中字典序第k小的数字
1
15 7
Sample Output
15
思路:完全十叉树
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
| #include <iostream> #include <cstring> #include <cstdlib> #include <algorithm> #include <queue> #include <cmath> #include <cstdio> #define mem(a,b) memset(a,b,sizeof(a))
using namespace std; typedef long long ll; const int MAXn=10005;
int cal(int n,int k){ int cur=1; k=k-1; while(k>0){ int step=0,first=cur,last=cur+1; while(first<=n){ step+=min(n+1,last)-first; first*=10; last*=10; } if(step<=k){ cur++; k-=step; } else{ cur*=10; k-=1; } } return cur; } int main() { int t; cin>>t; while(t--){ int n,k; cin>>n>>k; cout<<cal(n,k)<<endl; } return 0; }
|