【BZOJ 1005】[HNOI2008]明明的烦恼

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

 

数学相关

题解

我觉得我是做不出来了，只能学一下关于暴力分解的各种咯

#include<cstdio>
#define mod 100000
#include<cmath>
using namespace std;
int num[1001],ans[1001],pri[1001],d[1010];
int cnt,n,l,m,tot;
bool jud(int x){
     for(int i=2;i<=sqrt(x);i++)
         if(x%i==0)return 0;
     return 1;
}
 
void getpri(){
     for(int i=2;i<=1000;i++)
         if(jud(i))pri[++cnt]=i;
}
 
void solve(int a,int f){
    for(int i=2;i<=a;i++){
        int x=i;
        for(int j=1;j<=cnt;j++){
            if(x==0) break;
            while(x%pri[j]==0){
                num[j]+=f;x/=pri[j];
            }
        }
    }
}
 
void mult(int x){
    for(int i=1;i<=l;i++) ans[i]*=x;
    for(int i=1;i<=l;i++) ans[i+1]+=ans[i]/mod,ans[i]%=mod;
    while(ans[l+1]>0){
        l++;ans[l+1]=ans[l]/mod;ans[l]%=mod;
    }
}
void print()
{
    for(int i=l;i>0;i--)
        if(i==l)printf("%d",ans[i]);
        else printf("%05d",ans[i]);
}
int main(){
    getpri();
    int x;ans[1]=1;l=1;
    scanf("%d",&n);
    if(n==1){
        scanf("%d",&x);
        if(!x) {printf("1");return 0;}
        else {printf("0");return 0;}
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&d[i]);
        if(d[i]==0) {printf("0");return 0;}
        if(d[i]==-1) m++;
        else {d[i]--;tot+=d[i];} 
    }
    if(tot>n-2){printf("0");return 0;}
    solve(n-2,1);solve(n-2-tot,-1);
    for(int i=1;i<=n;i++) if(d[i]) solve(d[i],-1);
    for(int i=1;i<=n-2-tot;i++) mult(m);
    for(int i=1;i<=cnt;i++) while(num[i]--) mult(pri[i]);
    print();
}