【bzoj1005】[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

题解
prufer编码。
问题转化为prufer编码可能总数。

代码

#include<bits/stdc++.h>
#define N 1005
#define ll long long
#define mo 1000000007
using namespace std;
bool mark[1005];
int n,a[1005],ans[1000005],len;
int pri[1005],sum,tot,cnt,d[1005];
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void get_pri()
{
    for (int i=2;i<=1000;i++)
    {
        if (!mark[i]) pri[++tot]=i;
        for (int j=1;j<=tot&&i*pri[j]<=1000;j++)
        {
            mark[i*pri[j]]=0;
            if (i%pri[j]==0) break;
        }
    }
}
void modify(int x,int f)
{
    for (int k=2;k<=x;k++)
    {
        int t=k;
        for (int i=1;i<=tot&&pri[i]<=k;i++)
        {
            if (t<=1) break;
            int num=0;
            while (t%pri[i]==0)
            {
                num++;
                t/=pri[i];
            }
            a[i]+=f*num;
        }
    }
}
void mul(int x)
{
    int z=0;
    for (int i=1;i<=len;i++)
    {
        ans[i]=ans[i]*x+z;
        z=ans[i]/10;
        ans[i]%=10;
    }
    if (z)
    {
        while (z)
        {
            len++;
            ans[len]=z%10;
            z/=10;
        }
    }
}
int main()
{
    n=read();
    if (n==1)
    {
        int x=read();
        if (x==0) puts("1");
        else puts("0");
        return 0; 
    }
    get_pri();
    for (int i=1;i<=n;i++)
    {
        int x=read();
        if (x==0){puts("0");return 0;}
        if (x!=-1)
        {
            cnt++;
            d[cnt]=--x;
            sum+=x;
        }
    }
    if (sum>n-2){puts("0");return 0;}
    //cout<<"n-2="<<n-2<<endl;
    //cout<<"n-2-sum="<<n-2-sum<<endl;
    //cout<<"n-cnt="<<n-cnt<<endl;
    modify(n-2,1);modify(n-2-sum,-1);
    for (int i=1;i<=cnt;i++)
        modify(d[i],-1);
    int x=n-cnt;
    for (int i=1;i<=tot&&pri[i]<=x;i++)
    {
        if (x<=1) break;
        int num=0;
        while (x%pri[i]==0)
        {
            num++;
            x/=pri[i];
        }
        a[i]+=num*(n-2-sum);
    }
    len=1;ans[1]=1;
    for (int i=1;i<=tot;i++)
    {
        for (int j=1;j<=a[i];j++)
            mul(pri[i]);
    }
    for (int i=len;i;i--) printf("%d",ans[i]);
    puts("");
    return 0;
}