bzoj1005【hnoi2008】明明的烦恼

1005: [HNOI2008]明明的烦恼

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 3157   Solved: 1262
[ Submit][ Status][ Discuss]

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编码的有关性质：

①实现过程：不断删除书中度数为1的最小序号的编点，并输出与其相连的节点的编号，直至树中只剩两个节点。这样所形成的就是prufer编码。

②度数为d的节点在prufer编码中的出现次数为d-1。

③一颗节点数为n的树转化为的prufer编码中有n-2个数。

④将prufer编码还原成一棵树的过程：从prufer编码的最前端开始扫描节点，设该节点序号为u ，寻找不在prufer编码的最小序号且没有被标记的节点v，连接u和v，并标记v，将u从prufer编码中删除。扫描下一节点。

综上，我们可以发现prufer编码与树一一对应。所以这道题求满足条件的树的个数，也就是求满足条件的prufer编码的个数。

设m为无度数限制的节点个数，sum为有度数限制的节点的度数-1的和，即sum=∑(i=1,n) d[i]-1 ，(其中i要满足d[i]≥1)。

不考虑有度数限制的节点序号的重复，prufer编码个数为A(n-2,sum)。

有度数限制的节点序号的重复次数为∏(i=1,n) (d[i]-1)! ，(其中i要满足d[i]≥1)。

所以要满足有度数限制的节点，prufer编码个数为A(n-2,sum)/∏(i=1,n) (d[i]-1)! ，(其中i要满足d[i]≥1)。

无度数限制的节点序号可以任意的插在prufer编码中，个数为m^(n-2-sum)。

所以最终答案为 A(n-2,sum)*m^(n-2-sum)/∏(i=1,n) (d[i]-1)! ，(其中i要满足d[i]≥1)。

显然可以看出，这里要用到高精度。由于高精度除法比较难写，所以我们可以用质因数分解使运算更加简单。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
using namespace std;
int cnt,n,m,x,sum,l,ans[10001],f[1001],d[1001],p[1001];
void Prime()
{
	int a[1001];
	cnt=0;
	memset(a,0,sizeof(a));
	F(i,2,n) if (!a[i])
	{
		f[++cnt]=i;
		for (int j=i; j<=n; j+=i) a[j]=1;
	}
}
void Calc()
{
	l=1;
	memset(ans,0,sizeof(ans));
	ans[1]=1;
	F(i,1,cnt)
		F(j,1,p[i])
		{
			F(k,1,l) ans[k]*=f[i];
			F(k,1,l)
			{
				ans[k+1]+=ans[k]/10;
				ans[k]%=10;
			}
			if (ans[l+1]>0) l++;
			while (ans[l]>=10)
			{
				l++;
				ans[l]=ans[l-1]/10;
				ans[l-1]%=10;
			}
		}
}
void Print()
{
	D(i,l,1) printf("%d",ans[i]);
	printf("\n");

}
int main()
{
	scanf("%d",&n);
	sum=0; m=0;
	F(i,1,n)
	{
		scanf("%d",&d[i]);
		if (d[i]>0) sum=sum+d[i]-1;
		else m++;
	}
	Prime();
	memset(p,0,sizeof(p));
	F(i,n-1-sum,n-2)
	{
		x=i;
		if (x>1) F(j,1,cnt) while (x%f[j]==0)
		{
			x/=f[j];
			p[j]++;
		}
	}
	x=m;
	if (x>1) F(j,1,cnt) while (x%f[j]==0)
	{
		x/=f[j];
		p[j]=p[j]+n-2-sum;
	}
	F(i,1,n) F(j,1,d[i]-1)
	{
		x=j;
		if (x>1) F(k,1,cnt) while (x%f[k]==0)
		{
			x/=f[k];
			p[k]--;
		}
	}
	Calc();
	Print();
}