[bzoj1005][HNOI2008]明明的烦恼-Prufer编码+高精度

时间:2023-03-09 21:53:32
[bzoj1005][HNOI2008]明明的烦恼-Prufer编码+高精度

Brief Description

给出标号为1到N的点,以及某些点最终的度数,允许在

任意两点间连线,可产生多少棵度数满足要求的树?

Algorithm Design

结论题.

首先可以参考这篇文章了解一下什么是Prufer编码:

[bzoj1005][HNOI2008]明明的烦恼-Prufer编码+高精度

Cayley公式是说,一个完全图\(K_n\)有\(n^{n-2}\)棵生成树,换句话说n个节点的带标号的无根树有\(n^{n-2}\)个。今天我学到了Cayley公式的一个非常简单的证明,证明依赖于Prüfer编码,它是对带标号无根树的一种编码方式。

给定一棵带标号的无根树,找出编号最小的叶子节点,写下与它相邻的节点的编号,然后删掉这个叶子节点。反复执行这个操作直到只剩两个节点为止。由于节点数n>2的树总存在叶子节点,因此一棵n个节点的无根树唯一地对应了一个长度为n-2的数列,数列中的每个数都在1到n的范围内。下面我们只需要说明,任何一个长为n-2、取值范围在1到n之间的数列都唯一地对应了一棵n个节点的无根树,这样我们的带标号无根树就和Prüfer编码之间形成一一对应的关系,Cayley公式便不证自明了。

注意到,如果一个节点A不是叶子节点,那么它至少有两条边;但在上述过程结束后,整个图只剩下一条边,因此节点A的至少一个相邻节点被去掉过,节点A的编号将会在这棵树对应的Prüfer编码中出现。反过来,在Prüfer编码中出现过的数字显然不可能是这棵树(初始时)的叶子。于是我们看到,没有在Prüfer编码中出现过的数字恰好就是这棵树(初始时)的叶子节点。找出没有出现过的数字中最小的那一个(比如④),它就是与Prüfer编码中第一个数所标识的节点(比如③)相邻的叶子。接下来,我们递归地考虑后面n-3位编码(别忘了编码总长是n-2):找出除④以外不在后n-3位编码中的最小的数(左图的例子中是⑦),将它连接到整个编码的第2个数所对应的节点上(例子中还是③)。再接下来,找出除④和⑦以外后n-4位编码中最小的不被包含的数,做同样的处理……依次把③⑧②⑤⑥与编码中第3、4、5、6、7位所表示的节点相连。最后,我们还有①和⑨没处理过,直接把它们俩连接起来就行了。由于没处理过的节点数总比剩下的编码长度大2,因此我们总能找到一个最小的没在剩余编码中出现的数,算法总能进行下去。这样,任何一个Prüfer编码都唯一地对应了一棵无根树,有多少个n-2位的Prüfer编码就有多少个带标号的无根树。

一个有趣的推广是,n个节点的度依次为D1, D2, …, Dn的无根树共有\(\frac{(n-2)!}{\prod{(D_i-1)!}}\)个,因为此时Prüfer编码中的数字i恰好出现Di-1次。

然后, 根据排列组合和计数原理的相关知识, 设没有限制的节点数为\(m\), 答案就是

[bzoj1005][HNOI2008]明明的烦恼-Prufer编码+高精度

然后在分子和分母上分解因式, 把次数加加减减就好了.

Code

#include <algorithm>
#include <cctype>
#include <cstdio>
#include <cstring>
#define ll long long
const int maxn = 3000;
const ll mod = 1000000000000000;
inline int read() {
int x = 0, f = 1;
char ch = getchar();
while (!isdigit(ch)) {
if (ch == '-')
f = -1;
ch = getchar();
}
while (isdigit(ch)) {
x = x * 10 + ch - '0';
ch = getchar();
}
return x * f;
}
int n, m, tot, cnt;
ll d[maxn], num[maxn], prime[maxn], check[maxn], ans[maxn], l = 1;
void getprime() {
for (int i = 2; i <= 1000; i++) {
if (!check[i])
prime[cnt++] = i;
for (int j = 0; j < cnt; j++) {
if (i * prime[j] > 1000)
break;
check[i * prime[j]] = 1;
if (i % prime[j] == 0)
break;
}
}
}
void solve(int a, int f) {
for (int k = 1; k <= a; k++) {
int x = k;
for (int i = 0; i < cnt; i++) {
if (x <= 1)
break;
while (x % prime[i] == 0) {
num[i] += f;
x /= prime[i];
}
}
}
}
void mul(int x) {
for (int i = 1; i <= l; i++)
ans[i] *= x;
for (int i = 1; i <= l; i++) {
ans[i + 1] += ans[i] / mod;
ans[i] %= mod;
}
while (ans[l + 1]) {
l++;
ans[l + 1] += ans[l] / mod;
ans[l] %= mod;
}
}
void print() {
for (int i = l; i; i--)
if (i == l)
printf("%lld", ans[i]);
else
printf("%06lld", ans[i]);
}
int main() {
getprime();
ans[1] = 1;
n = read();
if (n == 1) {
int x = read();
if (!x || x == -1)
printf("1");
else
printf("0");
return 0;
}
for (int i = 1; i <= n; i++) {
d[i] = read();
if (!d[i]) {
printf("0");
return 0;
}
if (d[i] == -1)
m++;
else {
d[i]--;
tot += d[i];
}
}
if (tot > n - 2) {
printf("0");
return 0;
}
solve(n - 2, 1);
solve(n - 2 - tot, -1);
for (int i = 1; i <= n; i++)
solve(d[i], -1);
for (int i = 0; i < cnt; i++)
while (num[i]--)
mul(prime[i]);
for (int i = 1; i <= n - 2 - tot; i++)
mul(m);
print();
return 0;
}