bzoj 1211: [HNOI2004]树的计数

树的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;
}

 

转载于:https://www.cnblogs.com/lxy8584099/p/10432999.html