BZOJ 1005 明明的烦恼（Purfer序列+组合数学+数论+高精度）

Description
自从明明学了树的结构,就对奇怪的树产生了兴趣……给出标号为1到N的点,以及某些点最终的度数,允许在任意两点间连线,可产生多少棵度数满足要求的树?
Input
第一行为N(0 < N < = 1000),
接下来N行,第i+1行给出第i个节点的度数Di,如果对度数不要求,则输入-1
Output
一个整数,表示不同的满足要求的树的个数,无解输出0
Sample Input
3
1
-1
-1
Sample Output
2
Solution
介绍一下purfer序列，一个编号1~n的树与一个长度为n-2的purfer序列一一对应，下面看看两者如何一一对应
对于一棵1~n的树，用n-2步得到与其对应的purfer序列，第i步找到当前度数为1且编号最小的点u，将与其相连的点v的编号作为purfer序列的第i个元素，然后将u删去，做完n-2步我们得到一个长度为n-2的序列和一个只有两个节点的树，通过以上步骤可知一棵树只会得到一个唯一的purfer序列
对于长度为n-2的purfer序列，首先将1~n这n个点的度数初始化为1，然后再加上这个点的编号在purfer序列中出现的次数得到每个点的度数，之后做n-2步得到一棵树，第i步找到度数为1且编号最小的点u，然后取出序列第i个元素v，在树上构造一条u-v的边，然后将u,v的度数减一，重复以上过程得到一棵n个节点的树，通过以上步骤可知一个purfer序列对应一棵唯一的树
知道以上知识后我们再来看这道题，求合法的树的个数其实就变成求合法的purfer序列的个数，对于度数有限制的点i，假设有cnt个，我们知道其在purfer序列中出现的次数就是其度数a[i]减一，令，那么度数无限制的点在purfer序列中出现的次数之和为n-2-sum，每次可以选择剩余n-cnt个点中的任何一个点，故满足条件的purfer序列有个，当然这是在合法且n>2的情况下的计算方法，n为1和2时特判，不合法情况就是某个点度数为0或者大于n-1，还有sum>n-2的情况，对于合法情况，上面的结果也非常大，故要使用高精度乘法，为避免除法的出现，对于阶乘和组合数可以素因子分解后就变成若干素数的乘积了
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define mod 10000
#define maxn 1111
int ans[111111],len;
void mul(int n)
{
    int k=0;
    for(int i=0;i<len;i++)
    {
        int temp=ans[i]*n+k;
        ans[i]=temp%mod;
        k=temp/mod;
    }
    if(k!=0)ans[len]=k,len++;
}
void output()
{
    printf("%d",ans[len-1]);
    for(int i=len-2;i>=0;i--)
        printf("%04d",ans[i]);
    printf("\n");
}
int prime[maxn],res,mark[maxn];
void get_prime()
{
    memset(mark,0,sizeof(mark));
    res=0;
    for(int i=2;i<maxn;i++)
        if(!mark[i])
        {
            prime[res++]=i;
            for(int j=i;j<maxn;j+=i)mark[j]=1;
        }
}
int n,a[maxn],sum,cnt,num[maxn];
void deal(int n,int sta)
{
    for(int i=0;i<res&&prime[i]<=n;i++)
    {
        int m=n;
        while(m)
        {
            num[i]+=sta*(m/prime[i]);
            m/=prime[i];
        }
    }   
}
int main()
{
    get_prime();
    scanf("%d",&n);
    int gg=0;
    memset(num,0,sizeof(num));
    cnt=sum=0;
    len=1,ans[0]=1;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        if(a[i]==0||a[i]>=n)gg=1;
        if(a[i]==-1)continue;
        cnt++,sum+=a[i]-1;
    }
    if(n==1)
    {
        if(a[1]==-1||a[1]==0)printf("1\n");
        else printf("0\n");
        return 0;
    }
    if(n==2)
    {
        if((a[1]==-1||a[1]==1)&&(a[2]==-1||a[2]==1))printf("1\n");
        else printf("0\n");
        return 0;
    }
    if(gg||sum>n-2)
    {
        printf("0\n");
        return 0;
    }
    deal(n-2,1),deal(n-2-sum,-1);
    for(int i=1;i<=n;i++)
        if(a[i]!=-1)deal(a[i]-1,-1);
    for(int i=0;i<res;i++)
        while(num[i]--)mul(prime[i]);
    for(int i=1;i<=n-2-sum;i++)mul(n-cnt);
    output();
    return 0;
}