[bzoj1211] 树的计数 "prufer编码

[bzoj1211] 树的计数

---

prufer编码大概就是将一颗无根树对应到一个序列里面，然后构造就是不断找最小的度数为1的点然后将他的father加进去。

不难发现每个编号的出现次数就是deg+1

• 代码（特判无解写的极水，可能有锅）

#include<bits/stdc++.h>
using namespace std;
const int N=300;
typedef long long ll;
ll ans[N];
inline void modify(int P,int sum){
    for(int i=2;i<=P;i++){
        if(P==1)return;
        while(P%i==0){
            P/=i;ans[i]+=sum;
        }
    }
}
void calccomb(int a,int b){
    if(a==b) return;
    int x=a,y=b,z=a-b;
    for(int i=1;i<=x;i++)modify(i,1);
    for(int i=1;i<=y;i++)modify(i,-1);
    for(int i=1;i<=z;i++)modify(i,-1);
}
int n;
int tot=0;
int main()
{
    scanf("%d",&n);
    int P=n-2;
    for(int i=1;i<=n;i++){
        int D;scanf("%d",&D);
        if(D<=0&&n!=1){
            puts("0");
            return 0;
        }
        tot+=D;
        if(D>1)calccomb(P,D-1);
        P-=D-1;
    }
    if(tot!=(n-1)*2){
        puts("0");return 0;
    }
    ll ret=1;
    for(int i=2;i<=150;i++){
        while(ans[i]){
            ret*=(ll)i;
            ans[i]--;
        }
    }
    printf("%lld\n",ret);
}