首页 > 代码库 > BZOJ 1005 明明的烦恼 (组合数学)

BZOJ 1005 明明的烦恼 (组合数学)

题解:n为树的节点数,d[ ]为各节点的度数,m为无限制度数的节点数。

则            
 
所以要求在n-2大小的数组中插入tot各序号,共有种插法;
在tot各序号排列中,插第一个节点的方法有种插法;
插第二个节点的方法有种插法;
                                      .........
另外还有m各节点无度数限制,所以它们可任意排列在剩余的n-2-tot的空间中,排列方法总数为
 
根据乘法原理:
            
#include <cstdio>#include <cmath>int n,m,tot,i,j,d,down[1005],up[1005],p[1005],ans[10005];void pi(int x,int a[]){    for(int i=2;i<=x;i++)if(p[i]){        int sum=i; while(sum<=x)a[i]+=x/sum,sum*=i;    }}int main(){    scanf("%d",&n);    for(i=2;i<=1000;i++){        for(j=2;j<=std::sqrt(i);j++)        if(i%j==0)break;        if(j>sqrt(i))p[i]=1;    }    for(i=1;i<=n;i++){        scanf("%d",&d);        if(d==-1){m++;continue;}        if(d>1)pi(d-1,down);        tot+=d-1;    }    pi(n-2-tot,down);pi(n-2,up);    for(i=1;i<=1000;i++)up[i]-=down[i];    ans[0]=1;    for(i=1;i<=1000;i++)while(up[i]--){        for(j=0;j<=10000;j++)ans[j]*=i;        for(j=0;j<=10000;j++)if(ans[j]>9)ans[j+1]+=ans[j]/10,ans[j]%=10;    }    if(m)for(i=1;i<=n-2-tot;i++){        for(j=0;j<=10000;j++)ans[j]*=m;        for(j=0;j<=10000;j++)if(ans[j]>9)ans[j+1]+=ans[j]/10,ans[j]%=10;    }    if(tot>n-2||tot<n-2&&m==0)return puts("0"),0;     i=10000; while(!ans[i])i--;    while(~i)printf("%d",ans[i--]);    return 0;}