bzoj1005(prufer序列+组合+高精)

Description
自从明明学了树的结构,就对奇怪的树产生了兴趣…… 给出标号为1到N的点,以及某些点最终的度数,允许在任意两点间连线,可产生多少棵度数满足要求的树?
Input
第一行为N(0 < N < = 1000),接下来N行,第i+1行给出第i个节点的度数Di,如果对度数不要求,则输入-1
Output
一个整数,表示不同的满足要求的树的个数,无解输出0
Sample Input
3
1
-1
-1
Sample Output
2
HINT
两棵树分别为1-2-3;1-3-2

题解：
先介绍prufer序列：
（1）如何将树变为prufer序列：

不断删去度数为1的最小序号的点，并输出与其相连的节点的序号，直至树中只有2个节点。

（2）我们可以发现：

1.任意一棵n节点的树都可以唯一的用长度为n-2的prufer序列来表示。
2.度数为m的节点的序号在prufer序列中出现的次数为m-1。

（3）如何将prufer序列还原为一棵树：

从prufer序列的最前端开始扫描节点（u），寻找不在prufer序列的最小序号且没有被标记的节点v，连接u，v，并标记v，将u从prufer序列中删除。扫描下一节点。

该题运用了prufer序列的性质：
n为树的节点数，d[]为各节点的度数，sum为无限制度数的节点数。
则

tot=∑d[i]-1(i=1~n)
tot 即为有限制度数的节点在序列中出现的次数

所以要求在n-2大小的序列中插入tot个序号，共有C（n-2,tot）种插法；在tot个序号中，插第一个节点的方法有C（tot,d[1]-1）,插第二个节点的方法有C（tot-(d[1]-1),d[2]-1）;······
另外还有m个节点无度数限制，他们可以任意排列在剩余的n-2-tot的空间中，方案数为m^(n-2-tot);再根据乘法原理即可得解。
代码如下：

#include<cstdio>  
using namespace std;  
const int size=168;  
const int data[size+1]={0,2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,  
79,83,89,97,101,103,107,109,113,127,131,137,139,149,151,157,163,167,173,179,181,191,  
193,197,199,211,223,227,229,233,239,241,251,257,263,269,271,277,281,283,293,307,311,  
313,317,331,337,347,349,353,359,367,373,379,383,389,397,401,409,419,421,431,433,439,  
443,449,457,461,463,467,479,487,491,499,503,509,521,523,541,547,557,563,569,571,577,  
587,593,599,601,607,613,617,619,631,641,643,647,653,659,661,673,677,683,691,701,709,  
719,727,733,739,743,751,757,761,769,773,787,797,809,811,821,823,827,829,839,853,857  
,859,863,877,881,883,887,907,911,919,929,937,941,947,953,967,971,977,983,991,997};  
int n,len,i,a,sum,j,k;  
int t[10001],ans[size+1];  
void C(int n,int m)  
//在组合中用了质因数分解
{  
  int i,p,j;  
  for (i=m+1;i<=n;i++)  
  {  
    p=i;j=1;  
    while (p>1){while (p%data[j]==0) {ans[j]++;p/=data[j];}j++;}  
  }  
  for (i=1;i<=n-m;i++)  
  {  
    p=i;j=1;  
    while (p>1){while (p%data[j]==0) {ans[j]--;p/=data[j];}j++;}  
  }  
}  
int main()  
{  
  scanf("%d",&n);len=n-2;  
  for (i=1;i<=n;i++)  
  {  
    scanf("%d",&a);  
    if (a==0||len-(a-1)<0) {printf("0");return 0;}  
    if (a==-1) {sum++;continue;}a--;  
    C(len,a);len-=a;  
  }  
  if (len)  
  {  
    j=1;  
    while (sum>1){while (sum%data[j]==0) {ans[j]+=len;sum/=data[j];} j++;}  
  }  
  t[1]=1;k=1;  
  for (i=1;i<=size;i++)  
    while (ans[i])  
    {  
      ans[i]--;  
      for (j=1;j<=k;j++) t[j]*=data[i];  
      for (j=1;j<=k;j++) if (t[j]>9) {t[j+1]+=t[j]/10;t[j]%=10;}  
      while (t[k+1]) {k++;t[k+1]+=t[k]/10;t[k]%=10;}  
    }  
  for (i=k;i>0;i--)printf("%d",t[i]);  
  return 0;  
}