prufer BZOJ1211: [HNOI2004]树的计数

以前做过几题。。好久过去全忘了。

看来是要记一下。。。

【prufer】

n个点的无根树（点都是标号的，distinct）对应一个 长度n-2的数列

所以 n个点的无根树有n^（n-2）种

树 转 prufer数列：  每次删除编号最小的叶子节点，将与其相连的那个点 加入 prufer数列  直到树中只剩两个点，就结束

prufer数列 转 树：  首先是有个1到n的集合G，每次将prufer数列当前的第一项 和 当前G中 不在当前prufer里有的 最小的 元素x 连边。 接着删除当前prufer中的第一项 ，并在G中删除x。。直到prufer只剩两项，两者连边 结束

 

对树中的i号节点 在对应的prufer数列中 出现di-1次  （di为i号节点的度）

对于i号点度数为d[i]的 无根树 树的种数有 (n - 2) ! / ( (d1 - 1)! (d2 - 1)! ……(dn - 1)! )

1211: [HNOI2004]树的计数

所以 这是道基础题 上代码吧

#include <bits/stdc++.h>
using namespace std;
int n,d[160],a[160],b[160],c[160],t,k; long long x;
void hh(int x){
    for (int i=1;i<=t;++i){
        while (!(x%a[i])) x/=a[i],++b[i];
    }
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;++i) {
        scanf("%d",&d[i]);
        if ((d[i]<1&&n!=1)||d[i]>=n) {printf("0\n"); return 0;}
        x+=d[i];
    }
    if (x!=(n-1)*2) {printf("0\n"); return 0;}
    if (n<3) {printf("1\n");return 0;}
    sort(d+1,d+1+n);
    for (int i=1;i<=n;++i) --d[i];
    for (int i=2;i<=150;++i){ 
        k=1;
        for (int j=2;j<=i-1;++j)
        if (!(i%j)){k=0;break;}
        if (k) a[++t]=i;
    }
    k=1; while (!d[k]) ++k; while (d[k]==1) ++k;
    for (int i=2;i<=n;++i){
        hh(i); 
        while (d[k]==i){
            for (int j=1;j<=t;++j) c[j]-=b[j]; ++k;
        }
        if (n-2==i) for (int j=1;j<=t;++j) c[j]+=b[j];
    }
    x=1;
    for (int i=1;i<=t;++i)
    for (int j=1;j<=c[i];++j) x*=(long long)a[i];
    printf("%lld\n",x);
    return 0;
}