1005: [HNOI2008]明明的烦恼

Time Limit: 1 Sec  Memory Limit: 162 MB

Submit: 5090  Solved: 1986

[][][]

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序列+压位高精+组合数

黄学长无可挑剔题解

吓得我都不敢写了（其实是懒）

#include
        
         #define N 1001#define mod 1000000using namespace std;int a[N],tot,m,n;int prime[N],sum[N],cnt,ans[N];bool v[N];void pre_prime(){    for(int i=2;i
         
          N-1) break;            v[prime[j]*i]=true;            if(i%prime[j]==0) break;        }    }}void apart(int k,int w){    int tmp=k;    for(int j=1;j<=cnt;j++)    {         k=tmp;         while(k)         {             sum[j]+=w*k/prime[j];             k/=prime[j];         }    }}void mul(int x){    for(int i=1;i<=ans[0];i++) ans[i]*=x;    for(int i=1;i<=ans[0];i++)     {        ans[i+1]+=ans[i]/mod;        ans[i]%=mod;    }    while(ans[ans[0]+1])     {         ans[0]++;        ans[ans[0]+1]=ans[0]/mod;        ans[0]%=mod;     }}void out(){    printf("%d",ans[ans[0]]);    for(int i=ans[0]-1;i;i--) printf("%06d",ans[i]);}int main(){    scanf("%d",&n);    for(int i=1;i<=n;i++)     {        scanf("%d",&a[i]);        if(a[i]==-1) m++;        else tot+=--a[i];    }    if(tot>n-2) { printf("0"); return 0; }    pre_prime();    apart(n-2,1);    apart(n-2-tot,-1);    for(int i=1;i<=n;i++) if(a[i]) apart(a[i],-1);    //for(int i=1;i<=cnt;i++) printf("%d\n",sum[i]);    ans[0]=ans[1]=1;    for(int i=1;i<=cnt;i++)  for(int j=1;j<=sum[i];j++)  mul(prime[i]);    for(int i=1;i<=n-2-tot;i++) mul(m);    out();}