Luogu P2290 [HNOI2004]树的计数 Prufer序列+组合数

时间:2023-03-09 03:20:22
Luogu P2290 [HNOI2004]树的计数 Prufer序列+组合数

最近碰了$prufer$ 序列和组合数。。于是老师留了一道题:P2624 [HNOI2008]明明的烦恼

qwq要用高精。。。

于是我们有了弱化版:P2290 [HNOI2004]树的计数(考一样的可还行OvO)


首先前置知识:$Prufer序列$

然后,因为对于一个$ Prufer $序列有$n-2$ 项,而每个点的度数-1是这个点在$ Prufer$ 序列中出现的次数

所以。。。这不是多重集的排列吗(不懂多重集?

所以我们成功了一半(雾)

在计算时会爆$ long \space long $ 所以要拆分成质因数,然后去删除(反正不会T)

#include<cstdio>
#include<iostream>
#define ll long long
#define R register ll
using namespace std;
const int N=;
inline int g() {
R ret=,fix=; register char ch; while(!isdigit(ch=getchar())) fix=ch=='-'?-:fix;
do ret=ret*+(ch^); while(isdigit(ch=getchar())); return ret*fix;
}
int n,m,tot,cnt;
int r[N],num[N],pri[N];
ll fac[],ans=;
bool v[N];
inline void PRI() {
for(R i=;i<=N-;++i) {
if(!v[i]) pri[++cnt]=i;
for(R j=;j<=cnt&&i*pri[j]<=N-;++j) {
v[i*pri[j]]=true; if(i%pri[j]==) break;
}
}
}
inline void calc(ll x,int vl) {
for(R i=;i<=cnt;++i) {
if(x==) return ;
while(x%pri[i]==) num[i]+=vl,x/=pri[i];
}
}
signed main() { PRI(); fac[]=fac[]=;
for(R i=;i<=;++i) fac[i]=fac[i-]*i;
n=g(); if(n==) {
R x=g(); if(!x) printf("1\n");
else printf("0\n");
return ;
} for(R i=;i<=n;++i) {
r[i]=g(); if(!r[i]) {printf("0\n"); return ;}
--r[i]; tot+=r[i];
} if(tot!=n-) {printf("0\n"); return ;}
calc(fac[n-],); for(R i=;i<=n;++i) calc(fac[r[i]],-);
for(R i=;i<=cnt;++i) while(--num[i]>=) ans*=pri[i];
printf("%lld\n",ans);
}

2019.05.16