[BZOJ 1005][HNOI2008]明明的烦恼（prufer编码+组合数学+高精度）

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

HINT

---

两棵树分别为1-2-3;1-3-2

Solution

---

感觉其实和树的计数那题差不多，时隔四年…HNOI为何对prufer编码情有独钟0 0
算是加强版吧
cnt为度数未定的点的个数，sum为prufer数列中已确定的个数
答案是$C_{n-2}^{sum}\times\frac{sum!}{(d_1-1)\times(d_2-1)\times...\times(di-1)}\times{(n-2-sum)}^{cnt}$
即：在n-2个位置里选sum个位置放已确定的数×sum个位置的排列×剩余的位置可以在度数未定的点里随便选
一开始忘记乘$C_{n-2}^{sum}$了…
还是要分解质因数（可以避免掉除法）
要用高精度，不过是高精乘低精，好评

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#define MAXN 1002
using namespace std;
int n,sum=0,tot=0,cnt=0;
int d[MAXN],pri[MAXN],num[MAXN];
bool jud[MAXN];
int ans[100000],len;
void getprime()
{
    for(int i=2;i<=n;i++)
    {
        if(!jud[i])pri[++tot]=i;
        for(int j=1;j<=tot&&i*pri[j]<=n;j++)
        {
            jud[i*pri[j]]=1;
            if(i%pri[j]==0)break;
        }
    }
}
void add(int m,int n)
{
    int i=1;
    while(m>1)
    {
        while(m%pri[i]==0)
        m/=pri[i],num[i]+=n;
        i++;
    }
}
void mult(int x)
{
    int t=0;
    for(int i=0,g=0;i<len||g!=0;i++)
    {
        g+=ans[t]*x;
        ans[t]=g%10;
        g/=10;
        ++t;
    }
    len=t;
}
void calc()
{
    ans[0]=1,len=1;
    for(int i=1;i<=tot;i++)
    {
        for(int j=1;j<=num[i];j++)
        mult(pri[i]);
    }
    for(int i=1;i<=n-2-sum;i++)
    mult(cnt);
}
void print()
{
    for(int i=len-1;i>=0;i--)
    printf("%d",ans[i]);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&d[i]);
        if(d[i]==-1)
        {
            cnt++;
            continue;
        }
        if(!d[i])
        {
            printf("0");
            return 0;
        }
        sum+=d[i]-1;
    }
    if(sum>n-2)
    {
        printf("0");
        return 0;
    }
    getprime();
    for(int i=1;i<=sum;i++)add(i,1);
    for(int i=1;i<=n;i++)
    {
        if(d[i]!=-1)
        for(int j=1;j<=d[i]-1;j++)
        add(j,-1);
    }
    for(int i=1;i<=n-2;i++)add(i,1);
    for(int i=1;i<=sum;i++)add(i,-1);
    for(int i=1;i<=n-2-sum;i++)add(i,-1);
    calc();
    print();
    return 0;
}