bzoj1211 prufer编码

Description

一个有n个结点的树，设它的结点分别为v1, v2, …, vn，已知第i个结点vi的度数为di，问满足这样的条件的不同的树有多少棵。给定n，d1, d2, …, dn，编程需要输出满足d(vi)=di的树的个数。

Input

第一行是一个正整数n，表示树有n个结点。第二行有n个数，第i个数表示di，即树的第i个结点的度数。其中1<=n<=150，输入数据保证满足条件的树不超过10^17个。

Output

输出满足条件的树有多少棵。

Sample Input

4

2 1 2 1

Sample Output

2
HINT

Source

prufer编码的所有可能，用组合数学算。
(n - 2) ! / ( (d1 - 1)! (d2 - 1)! ……(dn - 1)! ) 虽然答案不会爆long long，但中间值也会爆的，所以要分解质因数来做

code:

#include <cstdio>  
#include <cstring>   
#include <string>  
#include <iostream>  
using namespace std;  
#define MAXN 500  
int cnt[MAXN]={0};  
int d[MAXN]={0};  
int n,sum;  

inline void cal(int x, int key)  
{  
    for (int i = 2; i <= x; i++)  
        if (x % i == 0)  
        {  
            while (x % i == 0 && x > 0)  
                cnt[i] += key, x /= i;  
        }  
}  

int main()  
{  
    scanf("%d", &n);  
    for (int i = 1; i <= n; i++)  
    {  
        scanf("%d", &d[i]);  
        sum += d[i];  
        if (d[i] == 0 && n > 1)  
        {  
            printf("0\n");  
            return 0;  
        }  
    }  
    if (sum != n * 2 - 2)  
    {  
        printf("0\n");  
        return 0;  
    }  
    if (n == 1)  
    {  
        printf("1\n");  
        return 0;  
    }  
    for (int i = 2; i <= n - 2; i++)  
        cal(i, 1);  
    for (int i = 1; i <= n; i++)  
        for (int j = 2; j <= d[i] - 1; j++)  
            cal(j, -1);  
    long long ans = 1;  
    for (int i = 1; i <= n; i++)  
        for (int j = 1; j <= cnt[i]; j++)  
            ans *= i;  
    printf("%lld\n", ans);  
}