bzoj 1211[HNOI2004]树的计数

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

ans=(n−2)!∏ni=1(di−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;}