1. 程式人生 > >bzoj 1211: [HNOI2004]樹的計數

bzoj 1211: [HNOI2004]樹的計數

urn 一個 sin amp code memset 序列 set 計數

樹的prufer序列

/**************************************************************
    Problem: 1211
    User: lxy8584099
    Language: C++
    Result: Accepted
    Time:8 ms
    Memory:1292 kb
****************************************************************/
 
/*
    簡單是prufer編碼問題
    每個節點i會出現di-1次 
    答案就是(n-2)! / π(di-1)!  (化簡後消去分子分母得到 
    索然答案不超過1e17 計算過程中會  所以分解質因子分子分母消去 
    需要考慮兩個特殊情況,
    第一個是只有一個結點,
    第二個是無法構成一個prufer數列。
*/ #include <bits/stdc++.h> using namespace std; typedef long long ll; const int INF=0x3f3f3f3f; const int maxn=155; int n; ll d[maxn]; ll cnt[maxn]; void makePrime(int x,int k){ int div=2; while(x>1){ if(x%div==0){ cnt[div]+=k; x/=div; }
else div++; } } int main(){ while(~scanf("%d",&n)){ int tot=0; for(int i=1;i<=n;i++){ scanf("%lld",&d[i]); if(d[i]>1) tot+=d[i]-1; } if(n==1) {printf("%d\n",!d[1]);continue;} if(tot!=n-2) {printf("0\n"
);continue;} memset(cnt,0,sizeof(cnt)); for(int i=2;i<=n-2;i++) makePrime(i,1); for(int i=1;i<=n;i++){ for(int j=2;j<=d[i]-1;j++){ makePrime(j,-1); } } ll ans=1; for(int i=1;i<=n;i++){ for(int j=1;j<=cnt[i];j++){ ans*=i; } } printf("%lld\n",ans); } return 0; }

bzoj 1211: [HNOI2004]樹的計數