bzoj1211: [HNOI2004]树的计数 prufer编码

时间:2021-11-06 16:22:31

题目链接

bzoj1211: [HNOI2004]树的计数

题解

prufer序

可重排列计数

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n = 0;
int b[10007];
int cnt[10007];
void Div(int x,int k = 1) {
for(int j = 2;j * j <= x;++ j) {
while(x % j == 0) {
cnt[j] += k;
x /= j;
}
}
cnt[x] += k;
}
main() {
int tot = 0;
scanf("%lld",&n);
for(int i = 1;i <= n;++ i) {
scanf("%lld",&b[i]);
if(!b[i] && n > 1) {
puts("0");return 0;
}
tot += b[i];
}
if(tot - n != n -2 ) {
puts("0"); return 0;
}
if(n <= 2) {puts("1"); return 0; }
for(int i = 2;i <= n - 2; ++ i) Div(i);
for(int i = 1;i <= n;++ i) {
for(int j = 2;j < b[i];++ j) Div(j,-1);
}
long long ans = 1;
for(int i = 1;i <= n;++ i)
for(int j = 1;j <= cnt[i];++ j) ans *= i;
cout << ans <<endl;
return 0;
}