[BZOJ1005]HNOI2008 明明的烦恼|prufer编码|排列组合

太可怕了，HN怎这么喜欢数学题，这几天被数学题虐成狗。。这题有一个什么prufer编码的东西，就是一棵树和一个prufer编码是唯一对应的，也就是把树的可能方案转化成一个数列的转化方案来做，然后就排列组合乱搞了，但是直接搞要写高精除，会死掉的。。显然方案数一定是整数，那么做一下质因数分解就OK了。。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<memory.h>
using namespace std;
struct gao{
int a[1000],len;
void print()
{
printf("%d",a[len]);
for (int i=len-1;i;i--)
{
if (a[i]/10==0){printf("000%d",a[i]);continue;}
if (a[i]/100==0){printf("00%d",a[i]);continue;}
if (a[i]/1000==0){printf("0%d",a[i]);continue;}
printf("%d",a[i]);
}
}
} ans;
int n,m,i,j,tot,npri=0,sum=0,cnt=0,p[1005],fz[1005],fm[1005],d[1005];
gao operator *(gao a,int k)
{
int i;
for (i=1;i<=a.len;i++) a.a[i]*=k;
for (i=1;i<=a.len;i++) a.a[i+1]+=a.a[i]/10000,a.a[i]%=10000;
for (i=a.len;;i++)
{
a.a[i+1]+=a.a[i]/10000,a.a[i]%=10000;
if (!a.a[i+1]){a.len=i;break;}
}
return a;
}
bool ispri(int x)
{
for (int i=2;i<=sqrt(x);i++) if (x%i==0) return false;
return true;
}
void getpri()
{
for (int i=2;i<=n;i++) if (ispri(i)) p[++npri]=i;
}
void dec(int *a,int k)
{
for (int i=1;i<=npri&&k>1;i++)
while (k%p[i]==0) a[i]++,k/=p[i];
}
int main()
{
freopen("1005.in","r",stdin);
freopen("1005.out","w",stdout);
scanf("%d",&n);
getpri();
for (i=1;i<=n;i++)
{
scanf("%d",&d[i]);
if (!d[i]&&n!=1){printf("0");return 0;}
if (d[i]!=-1) cnt++,sum+=--d[i];
}
if (sum>n-2) {printf("0");return 0;}
memset(fz,0,sizeof(fz));memset(fm,0,sizeof(fm));
dec(fz,n-cnt);
for (i=1;i<=npri;i++) fz[i]*=n-2-sum;
for (i=n-1-sum;i<=n-2;i++) dec(fz,i);
for (i=1;i<=cnt;i++)
for (j=1;j<=d[i];j++) dec(fm,j);
memset(ans.a,0,sizeof(ans.a));
ans.len=1;ans.a[1]=1;
for (i=1;i<=npri;i++)
{
fz[i]-=fm[i];
for (j=1;j<=fz[i];j++) ans=ans*p[i];
}
ans.print();
fclose(stdin);
fclose(stdout);
}