……ZOJ 1062
题目链接:………………
对于我来说,这真的是一道毒瘤题……还有就是买了一本书,第二章就给我来了这道破题,不过还好有收获
卡特兰数,一直有看到,无论是一些博客上,还是算法紫书上,但是对于数论我一般都是无力的……
题解
设 h [ x ] h[x] h[x]表示有 i i i 个节点的树可能的形态数
则 h [ 0 ] = 0 , h [ 1 ] = 0 h[0] = 0, h[1] = 0 h[0]=0,h[1]=0;对于 h [ 2 ] h[2] h[2],我们可以考虑,这棵数可以构造为左儿子中节点数为 0 0 0 且右儿子节点数为 1 1 1 的子树,或者左儿子中节点数为 1 1 1 且右儿子节点数为 0 0 0 的子树
故 h [ 2 ] = h [ 0 ] ∗ h [ 1 ] + h [ 1 ] ∗ h [ 0 ] = 2 h[2] = h[0] * h[1] + h[1] * h[0] = 2 h[2]=h[0]∗h[1]+h[1]∗h[0]=2
不难推广一下,对于有 k k k 个节点的二叉树,我们设其左儿子中有 i i i 个节点( i < k i < k i<k),则右儿子中的节点个数为( k − 1 − i k - 1 - i k−1−i),则该种情况下的可能的二叉树形态数为 h [ i ] ∗ h [ k − 1 − i ] h[i] * h[k - 1 - i] h[i]∗h[k−1−i]
则令
i
i
i 取所有可能取的值,乘法原理 + 加法原理可得:
h
[
n
]
=
∑
i
=
0
n
−
1
h
[
i
]
×
h
[
n
−
1
−
i
]
h[n] = \sum_{i = 0}^{n - 1} h[i] \times h[n - 1 - i]
h[n]=∑i=0n−1h[i]×h[n−1−i]
这个数列就是卡特兰数列(还有很多组合问题会用到他………………)
OK,求得了任意节点的二叉树的形态数后,我们可以根据这个来判断编号为 N N N 的树有几个节点(题目输入没有提供这个信息)
首先,我们来看一看 k k k 个节点的二叉树中,排名第一的树的编号为多少
- k = 1时,编号为 1
- k = 2时,编号为 2
- k = 3时,编号为 4
- k = 4时,编号为 9
求
h
[
n
]
h[n]
h[n] 前缀和
s
u
m
[
n
]
=
sum[n] =
sum[n]= {
1
,
2
,
4
,
9
,
.
.
.
.
.
.
1,2,4,9,......
1,2,4,9,...... } (忽略
s
u
m
[
0
]
sum[0]
sum[0])
可以发现规律:如果一棵二叉树的编号为N,则当
s
u
m
[
i
]
<
N
<
=
s
u
m
[
i
+
1
]
sum[i] < N <= sum[i + 1]
sum[i]<N<=sum[i+1]时,这棵树有
i
i
i 个节点
直观理解:前面节点数小于 i i i 的树的每一种形态,占一个编号,则对于一个 k k k 个节点的树,他在前面占用的编号的个数为 h [ k ] h[k] h[k],所以如果 N > = 前 i 棵 树 占 用 的 所 有 编 号 数 N >= 前 i 棵树占用的所有编号数 N>=前i棵树占用的所有编号数,说明 N N N 不可能只有小于 i i i 个节点
求得了这棵树中有 i i i 个节点,我们用 N N N 减去 s u m [ i − 1 ] sum[i - 1] sum[i−1],即减去节点数小于 i i i 的树占用的编号数,得到的便是当前要计算的二叉树(结果)在有 i i i 个节点的树中的排名。
现在我们要做的就是算出左儿子的节点数
l
s
ls
ls 和右儿子的节点数
r
s
rs
rs ,而已知的就是排名
不过我们知道排名从
1
1
1 到
h
[
0
]
×
h
[
i
−
1
]
h[0] \times h[i - 1]
h[0]×h[i−1]的树为左儿子0个节点,右儿子i - 1个节点的所有可能形态,接下来
h
[
1
]
×
h
[
i
−
2
]
h[1] \times h[i - 2]
h[1]×h[i−2] 棵树是左儿子1个节点,右儿子i - 2个节点的所有可能形态……
所以我们用 N N N依次减去(注意,N = N - XXX) h [ 0 ] × h [ i − 1 ] , h [ 1 ] × h [ i − 2 ] , … … , h [ l s ] × h [ i − 1 − l s ] h[0]\times h[i - 1],h[1]\times h[i - 2],……,h[ls]\times h[i - 1 - ls] h[0]×h[i−1],h[1]×h[i−2],……,h[ls]×h[i−1−ls],一直到减不了(结果 < 0)为止,此时 l s ls ls 就是左儿子拥有的节点数,令 r s = i − 1 − l s rs = i - 1 - ls rs=i−1−ls 为右儿子的节点数
接下来……计算左右子树在其拥有的节点数的二叉树中的排名
因为左右子树的变化是一个组合关系,所以每次我们枚举完所有右子树的所有形态后,左子树形态才会改变一次
这里和我们数数类似,我们各位数完0~9后,十位才变1个;我们数完99个数后,百位才变一个,与上面一个道理
此时我们已经得到了左右儿子的节点数,此时 N N N已经减去了一堆东西,此时的 N N N代表了:有 i i i 个节点的二叉树,且左右子树的节点数分别为 l s , r s ls,rs ls,rs 时树的排名
结合上面的例子,将 N N N 整除 h [ r s ] h[rs] h[rs] ,得到的商就是左子树的排名,而余数就是右子树的排名,这里我觉得的确会有些难理解、、、
剩下的一些细节,我会写明在注释中……
code
#include <iostream>
#include <cstdio>
using namespace std;
int h[20], s[20]; // 卡特兰数 和 前缀和
void init () {
h[0] = 1, h[1] = 1;
s[1] = 0, s[1] = 1;
for (int i = 2;i <= 19; ++ i) {
int ctl = 0;
int k = i - 1;
for (int j = 0;j < i; ++ j, -- k) { // 计算卡特兰数 列
ctl += h[j] * h[k];
}
h[i] = ctl;
s[i] = s[i - 1] + ctl;
}
}
void DFS (int n, int x) { // x 从 1 开始
if (n == 1) {
printf ("X");
return;
}
if (n == 0) return;
// 以上就不用解释了
int i, j, left;
j = n - 1;
for (i = 0; i < n; ++ i, -- j) {
left = h[i] * h[j];
if (x > left) x -= left; // 计算出左、右儿子的节点数
else break;
}
// 此时 x 等于 n 个节点 , 左儿子 有 i 个节点的第 x 棵树
x -= 1; // mod运算针对于0 ~ (n-1), 而这里的 x 从 1 开始
if (i != 0) {
printf ("(");
DFS (i, (x / h[j]) + 1); // 但是我们还是要把编号转到从 1 开始的
//不习惯的可以改一下代码
printf (")");
}
printf ("X");
if (j != 0) {
printf ("(");
DFS (j, (x % h[j]) + 1);
printf (")");
}
}
int main () {
init ();
int n;
while (scanf ("%d", &n) && n) {
int i;
for (i = 1;i <= 19; ++ i) {
if (n < s[i]) break;
}
DFS (i, n - s[i - 1]); // n - s[i - 1] 代表这是 i - 1 个节点的树中的第 n - s[i - 1] 棵树
}
return 0;
}