BZOJ 1211 树的计数

http://www.lydsy.com/JudgeOnline/problem.php?id=1211

思路:每一个prufer编码都代表了一棵树，而点的度数，代表了它在prufer编码中出现的次数+1，因此，我们就知道每个点在prufer编码中出现的次数，用组合数就可以解决:

最后的式子约一下是: cnt!/(∏(du[i]-1)!)

要分解质因数，不然会爆longlong

#include<cstdio>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<cstring>
#define ll long long
ll s[];
int p[],n,du[],num[];
int read(){
    int t=,f=;char ch=getchar();
    while (ch<''||''<ch){if (ch=='-') f=-;ch=getchar();}
    while (''<=ch&&ch<=''){t=t*+ch-'';ch=getchar();}
    return t*f;
}
bool jud(int x){
    for (int i=;i<=sqrt(x);i++)
     if (x%i==) return ;
    return ;
}
void init(){
    for (int i=;i<=;i++)
     if (jud(i)) p[++p[]]=i;
}
void solve(ll x,int v){
    if (x==) return;
    for (int i=;i<=p[];i++){
        if (x<=) return;
        while (x%p[i]==){
            num[i]+=v;
            x/=p[i];
        }
    }
}
int main(){
    n=read();
    s[]=;
    s[]=;
    for (int i=;i<=;i++) s[i]=s[i-]*i;
    init();
    ll ans=,tot=;
    for (int i=;i<=n;i++){
     du[i]=read(),du[i]--,tot+=du[i];
    }
    if (n==){
        if (tot==-) puts("");
        else puts("");
        return ;
    }
    for (int i=;i<=n;i++){
        if (du[i]<) {
            puts("");
            return ;
        }
    }
    if (tot!=n-){
        puts("");
        return ;
    }
    solve(s[n-],);
    for (int i=;i<=n;i++)
     solve(s[du[i]],-);
    for (int i=;i<=p[];i++)
     for (int j=;j<=num[i];j++)
      ans*=p[i];
    printf("%lld\n",ans);
    return ;
}