zyb的面试（hdu 6468 十叉树）

题目链接：

zyb的面试

 

题意：

一个 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;
}