BZOJ1005: [HNOI2008]明明的烦恼(purfer编码)

  题意：传送门
  题解：参考源自传送门
  一个含有 n 个节点的 $PurferSequence$ 有 $n-2$ 个数， $PurferSequence$ 中的每个数是 $1\sim n$ 中的一个数
  一个定理：一个$PurferSequence$和一棵树一一对应
  首先了解由一个树得到 $PurferSequence$，由一棵树得到它的 $PurferSequence$ 总共需要 $n-2$ 步，每一步都在当前的树中寻找具有最小标号的叶子节点（度为 $1$），将与其相连的点的标号设为 $PurferSequence$ 的第 $i$ 个元素，并将此叶子节点从树中删除，直到最后得到一个长度为 $n-2$ 的 $PurferSequence$ 和一个只有两个节点的树，具体例子参考上述博客。
  其次需要了解由 $PurferSequence$ 得到一棵树，先将所有编号为 $1$ 到 $n$ 的点的度赋初值为 $1$，然后加上它在 $PurferSequence$ 中出现的次数，得到每个点的度，先执行 $n-2$ 步，每一步，选取具有最小标号的度为 $1$ 的点 $u$ 与 $PurferSequence$ 中的第 $i$ 个数 $v$ 表示的顶点相连，得到树中的一条边，并将 $u$ 和 $v$ 的度减一，最后再把剩下的两个度为 $1$ 的点连边，加入到树中，具体例子参考上述博客。
  有了上述这些知识后，看下这个题如何做呢？每个树加了度数限制，先不考虑无解的情况，从 $PurferSequence$ 构造树的过程中可知，一个点的度数减一表示它在 $PurferSequence$ 中出现了几次，那么：
  假设度数有限制的点的数量为 $cnt$，他们的度数分别为：$d[i]$，设$$sum=\sum _{i=1}^{cnt}(d[i]-1)$$
  那么，在 Purfer Sequence 中的不同排列的总数为：
$$C_{n-2}^{sum}\ast \frac{sum!}{{\prod }_{i=1}^{cnt}(d[i]-1)!}$$
  而剩下的 n-2-sum 个位置，可以随意的排列剩余的 n-cnt 个点，于是，总的方案数就应该是：
$$C_{n-2}^{sum}\ast \frac{sum!}{{\prod }_{i=1}^{cnt}(d[i]-1)!}\ast (n-cnt{)}^{n-2-sum}$$
  化简之后为：
$$\frac{(n-2)!}{sum!\ast {\prod }_{i=2}^{cnt}(d[i]-1)!}\ast (n-cnt{)}^{n-2-sum}$$
  判断有解和无解其实也是非常容易的，不能构成 $PurferSequence$ 编码的无非是 $0$ 或者 $1$ 的情况或者其他点数的情况，这两个可以直接通过度数判断，但是能构成 $PurferSequence$ 的情况下，有的点数超过了 $n-1$ 或者等于了 $0$，这样肯定也是构不成这样的树的。

import java.util.*;
import java.math.*;
 
public class Main {
    static int n, d[]=new int[10002];
    static BigInteger p[]=new BigInteger[1002];
    static BigInteger ans;
     
    static public void main(String args[]) {
        Scanner IN=new Scanner(System.in);
        n=IN.nextInt();
        int sum=0, flag=0, cnt=0;
        for(int i=0; i<n; i++) {
            d[i]=IN.nextInt();
            if(d[i]==0 || d[i]>n-1) flag=1;
            if(d[i]==-1) continue;
            sum+=d[i]-1;
            cnt++;
        }
        IN.close();
        if(n==1) {
            if(d[0]==0 || d[0]==-1) System.out.println(1);
            else System.out.println(0);
            return;
        }
        if(n==2) {
            if((d[0]==-1 || d[0]==1) && (d[1]==-1 || d[1]==-1)) System.out.println(1);
            else System.out.println(0);
            return;
        }
        if(flag==1) System.out.println(0);
        p[0]=BigInteger.ONE;
        for(int i=1; i<=n; i++) p[i]=p[i-1].multiply(BigInteger.valueOf(i));
        ans=p[n-2].divide(p[n-2-sum]);
        for(int i=0; i<n; i++) {
            if(d[i]==-1) continue;
            ans=ans.divide(p[d[i]-1]);
        }
        for(int i=0; i<n-2-sum; i++) ans=ans.multiply(BigInteger.valueOf(n-cnt)); 
        System.out.println(ans);
    }
}