题目链接:
题意:
一个 1-n 的排列,按字典序排序,小的在前,求第 k 个数的值。(例(n=15):1, 10, 11, 12, 13, 14, 15, 2, 3, 4, 5, 6, 7, 8, 9)
思路:
建立一个十叉树,如下图所示:
该树满足如下性质:
1. 不算最后一层,均为满十叉树。
2. 先序遍历到的第 k 个数就是所求字典序第 k 小的数。
3. 一个点到其最左的子节点步数+1,到其右边第一个兄弟节点,步数+该点的子树大小。
根据这些性质即可进行求解:
每个节点的子树大小可以在O(1)时间内算得。若该节点子树中的最大步数>=k,就往下走(走到其最左的子节点),<k就往右走(走到其右边第一个兄弟节点),它们的步数按性质3计算。若该节点的步数恰好等于 k ,输出即可。
Code:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX = 1e6 + 20;
const int mod = 123456789;
int n, k;
ll f[20], sum[20];
int lenb;
void dfs(int root, int num)
{
if (num == k) {
printf("%d\n", root);
return;
}
//计算字数大小siz
int siz = 1;
int lena = log10((double)root) + 1;
if (lena + 1 <= lenb) {
siz = 0;
siz += (sum[lenb - 1 - lena]);
if (min(1ll * n, (root + 1)*f[lenb - lena] - 1) >= root * f[lenb - lena])
siz += min(1ll * n, (root + 1)*f[lenb - lena] - 1) - root * f[lenb - lena] + 1;
}
if (num + siz - 1 >= k) {
dfs(root * 10, num + 1);
}
else {
dfs(root + 1, num + siz);
}
return;
}
int main()
{
int T;
scanf("%d", &T);
f[0] = 1;
for (int i = 1; i < 15; i++) {
f[i] = f[i - 1] * 10;
}
sum[0] = f[0];
for (int i = 1; i < 15; i++) {
sum[i] = sum[i - 1] + f[i];
}
while (T--)
{
scanf("%d%d", &n, &k);
lenb = log10((double)n) + 1;
dfs(1, 1);
}
return 0;
}