BZOJ1211 - [HNOI2004]树的计数

Portal

Description

一个有\(n(n\leq150)\)个结点的树,给出每个节点的度数\(d_i\),求满足度数要求的无根树的个数。

Solution

prufer数列_百度百科
简单来说,一棵\(n\)个节点的无根树与一个数值在\([1,n]\)的长度为\(n-2\)的序列一一对应,这个序列便称作prufer数列。其中若点\(i\)的度数为\(d_i\),那么\(i\)在prufer数列中就出现\(d_i-1\)次。
那么这道题就很简单了:向一个长度为\(n-2\)的序列中填入\(d_1-1\)\(1\)\(d_2-1\)\(2\),...,\(d_n-1\)\(n\)

\[\begin{align*} ans &= \binom{n-2}{d_1-1} \binom{n-2-(d_1-1)}{d_2-1} ...\binom{n-2-\sum_{i=1}^{n-1}(d_i-1)}{d_n-1} \\ &= \frac{(n-2)!}{(d_1-1)!(n-2-(d_1-1))!} \cdot \frac{(n-2-(d_1-1))!}{(d_2-1)!(n-2-\sum_{i=1}^2(d_i-1))} ... \frac{(n-2-\sum_{i=1}^{n-1}(d_i-1))!}{(d_n-1)!0!} \\ &= \frac{(n-2)!}{\prod_{i=1}^n (d_i-1)!} \end{align*}$$由于阶乘比较大而且坑爹的不取模,所以通过分解质因数的方法来计算。 > 时间复杂度$O(n^2)$。 ##Code ```cpp //[HNOI2004]树的计数 #include <cstdio> int const N=200; int n,d[N]; int fac[N][N]; void init() { for(int i=1;i<=n;i++) { int x=i; for(int j=2;j<=n;j++) while(x%j==0) fac[i][j]++,x/=j; } for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) fac[i][j]+=fac[i-1][j]; } int ansP[N]; int main() { scanf("%d",&n); init(); int sum=0; for(int i=1;i<=n;i++) { scanf("%d",&d[i]); sum+=d[i]; if(d[i]==0&&n!=1) {puts("0"); return 0;} } if(sum!=n*2-2) {puts("0"); return 0;} if(n==1) {puts("1"); return 0;} for(int i=1;i<=n;i++) ansP[i]=fac[n-2][i]; for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) ansP[j]-=fac[d[i]-1][j]; long long ans=1; for(int i=1;i<=n;i++) for(int j=1;j<=ansP[i];j++) ans*=i; printf("%lld\n",ans); return 0; } ``` ##P.S. 注意要特判掉无解和$n=1$的情况。\]

posted @ 2018-03-27 20:50  VisJiao  阅读(123)  评论(0编辑  收藏  举报