【bzoj1284】【HNOI2004】【树的计数】【组合数学+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

题解：首先需要知道有个东西叫prufer数列。然后可以发现一个节点在prufer数列中出现的次数是这个节点的度数减一。这样我们就知道这个数列中有哪些数了。因为一个prufer数列唯一对应一颗树。然后问题就变成了求有多少种prufer数列。又因为我们知道了元素种类与出现次数。于是问题就变成了求一个有重复元素的全排列。

因为n最大有150.所以分解一下质因数就好了。

注意一开始要先判断是否有解。即把所有点的度数减一然后加起来，如果不等于n-2就无解。

#include<cstdio>
using namespace std;
int n,tot,p[2][151],x;
long long ans(1);
void cal(int x,int k)
{   
  for (int i=2;;i++)
     {
        if (x%i==0) while (x%i==0) {p[k][i]++;x/=i;}
        if (x==1) break;
     }
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) 
     {
       scanf("%d",&x);
       if (x>2) 
        {
          for (int j=1;j<=x-1;j++) cal(j,1);
        }
       if (x>1) tot+=x-1; 
     }
    if (n==1) {printf("%d",!x);return 0;}
    if(tot!=n-2) {printf("0");return 0;} 
    for (int i=2;i<=tot;i++) cal(i,0);
    for (int i=1;i<=150;i++)
     if (p[0][i]-p[1][i]>0)
       for (int j=1;j<=p[0][i]-p[1][i];j++)
         ans*=(long long)i;
    printf("%lld",ans);
}