bzoj 1211[HNOI2004]树的计数

题目链接
prufer序列可以知道度数为d的点会出现d-1次,而一共有n-2 个数出现,
显然可以由排列公式得知:

$$ans=\frac{(n - 2)!}{\prod_{i = 1}^{n}{(d_i - 1)!}}$$
Ps:乘法中会爆掉long long可以用唯一分解定理表示 我会说这样预处理,我跑了bzoj倒数第二!!!,也可以用多个数的乘积表示.注意对不成立情况的讨论

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const ll N=1e6+10;
vector<int>pri;vector<int>tm; 
bool check[N];ll d[200],n,tot=0,flag=1;
void getprime()
{
    memset(check,0,sizeof(check));
    for(int i=2;i<=N;i++) {
        if(!check[i]) pri.push_back(i),tm.push_back(0);
        for(int j=0;j<int(pri.size());j++) {
            if(pri[j]*i>N) break;
            check[i*pri[j]]=1;
            if(i%pri[j]==0) break;
        }
    }
}
void Dec(ll x,int data)
{
    if(x==0||x==1) return ;
    for(int i=0;i<int(pri.size());i++) {
        if(x==1) break;
        while(x%pri[i]==0) {
            x/=pri[i];
            tm[i]+=data;
        }
    } 
    return ;
}
ll qpow(ll a,ll b)
{
    ll ret=1;
    for(ll i=b;i;i>>=1,a*=a) 
        if(i&1) ret*=a;
    return ret;
}
ll solve()
{
    if(tot!=n-2||!flag||(n==1&&d[1]==1)) return 0;
    for(int i=1;i<=n-2;i++) 
            Dec(i,1);       
    for(int i=1;i<=n;i++) {
        for(int j=d[i]-1;j>=0;j--) {
            Dec(j,-1);
        }
    }
    ll ans=1;
    for(int i=0;i<int(pri.size());i++) 
        ans*=qpow(pri[i],tm[i]);
    return ans;
}
int main()
{
    getprime();
    scanf("%lld",&n);
    fo(i,1,n) {scanf("%lld",&d[i]);tot+=d[i]-1;if(!d[i]&&n!=1) flag=0;}
    printf("%lld\n",solve());
    return 0;
}