[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);
}