[BZOJ1211] [HNOI2004] 树的计数 [Prüfer编码][Caylay公式]

Link
Luogu - https://www.luogu.org/problemnew/show/P2290
BZOJ - https://www.lydsy.com/JudgeOnline/problem.php?id=1211

---

参考

---

定义：无根树的叶子结点满足：度数为 1 。

求Prüfer编码：每次取出编号最小的叶子放到编码序列尾端，删除这个叶子
用一个平衡树/stl set，$O(n\log n)$

还原无根树：初始化 $u$ 为 Prüfer 序列， $v$ 为点集。
每次取出 $u$ 中第一个点 $x$，和 $v$ 中编号最小的不在 $u$ 中的点 $y$ 连边，在 $u,v$ 中分别删去 $x,y$ 。
用一个平衡树/stl set，$O(n\log n)$

---

一棵 $n\text{n}n$ 点无根树唯一对应一个长度 $n-2\text{n−2}n-2$ ，每个元素的值 $\in [1,n]\text{∈[1,n]}\in [1,n]$ 的数列

引理 1 - 有标号无根树计数 $n^{n-2}$
引理 2 - 度数为 $i$ 的点会在 Prüfer 序列中出现 $i-1$ 次

---

然后结论就非常显然了。
这道题好像要高精度啊
能不能绕过去？
唯一分解，指数相减。

---

这题数据好弱啊 我也不知道我的代码是对是错
加强版HNOI2008那道题
不过反正都挺纸张的
人生苦短 我再见

---

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int n;
long long Ans = 1;
int Mark[1005];
int Extt[1005];
int Anss[1005];
bool NotPrime[1005];
int PrimeList[1005];
int Cor[1005];
void Red()
{
	NotPrime[0]=NotPrime[1]=1;
	for (register int i = 2; i <= n; ++i)
	{
		if(!NotPrime[i]) PrimeList[++PrimeList[0]] = i, Cor[i] = PrimeList[0];
		for (register int j = 1, t = i * PrimeList[j]; j <= PrimeList[0] && t <= n; ++j)
		{
			NotPrime[t] = 1;
			if (!(i % PrimeList[j])) break;
		}
	}
}
void pre()
{
	for (register int t, i = 2; i <= n; ++i)
	{
		t = i;
		for (register int j = 1; j <= PrimeList[0] && PrimeList[j] * PrimeList[j] <= t; ++j)
		{
			while (t % PrimeList[j] == 0) t /= PrimeList[j], ++Extt[j];
		}
		if (t != 1) ++Extt[Cor[t]];
		for (register int j = 1; j <= PrimeList[0]; ++j) Anss[j] += Mark[i] * Extt[j];
	}
}
long long qpow(int a, int b)
{
	long long ret = 1;
	while (b)
	{
		if (b & 1) ret *= a;
		a *= a;
		b >>= 1;
	}
	return ret;
}
int Sum;
int main()
{
	bool failans = 0;
	bool fail = 0;
	scanf("%d", &n);
	++Mark[n-2];
	Red();
	for (register int degree, i = 1; i <= n; ++i)
	{
		scanf("%d", &degree);
		if(fail)continue;
		if (degree >= n) {fail=1,failans = 0;continue;}
		if (n == 1) 
		{
			fail=1;
			if (degree == 0) failans = 1;
			else failans = 0;
			continue;
		}
		if (degree == 0) {fail=1,failans = 0;continue;}
		--Mark[degree-1];
		Sum += degree;
	}
	if (Sum != 2 * (n - 1)) fail=1,failans = 0;
	if(fail)return printf("%d",failans),0;
	pre();
	for (register int i = 1; i <= PrimeList[0]; ++i) Ans *= qpow(PrimeList[i], Anss[i]);
	printf("%lld", Ans);
	return 0;
}