bzoj 1005: [HNOI2008]明明的烦恼（prufer数列）

1005: [HNOI2008]明明的烦恼

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 5171   Solved: 2021
[ 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

根据数的度数求数的种类可以用prufer数列

prufer数列是无根树的一种数列。在组合数学中，Prufer数列由有一个对于顶点标过号的树转化来的数列，点数为n的

树转化来的Prufer数列长度为n-2。它可以通过简单的迭代方法计算出来

一种生成Prufer序列的方法是迭代删点，直到原图仅剩两个点。对于一棵顶点已经经过编号的树T，顶点的编号为{1,2,...,n}，在第i步时，移去所有叶子节点（度为1的顶点）中标号最小的顶点和相连的边，并把与它相邻的点的编号加入Prufer序列中，重复以上步骤直到原图仅剩2个顶点。

例子

以下面的树为例子，首先在所有叶子节点中编号最小的点是2，和它相邻的点的编号是3，将3加入序列并删除编号为2的点。接下来删除的点是4，5被加入序列，然后删除5，1被加入序列，1被删除，3被加入序列，此时原图仅剩两个点（即3和6），Prufer序列构建完成，为{3,5,1,3}

性质

1：任意一点的度为d，那么这个数一定会在这个序列中存在d-1个

2：序列和树一一对应

这题我们假设度数已知，第i个点的度数为di，那么我们可以构造出的数列个数（树的个数）就为

可这题我们有些点的度数并不知道，假设我们已知cnt个点的度数，它们的度数之和为sum+cnt，因为序列长度为

n+2，所以无视空位可以构造出的数列个数（树的个数）就为

（其中所有的di都是已知的点的度数）

因为还有n-cut个位置，剩下的每个位置可以放任意一个未知度数的点，所以这题答案就是

冷静的化简下↓

但还没那么简单，很显然这题的答案是个超大的数，需要用到高精度乘法，可由于有分母，所以要将每个元素

都分解质因数，分子的质因数和分母的公共质因数约掉之后再×就好了

注意特判n=1和n=2以及不合法的情况（sum过大或者某个度节点数过大或为0）

#include<stdio.h>
#include<string.h>
int k, a[1005], pri[1005], cot[1005], ans[10005], p[1005] = {1,1};
void Add(int n, int m)
{
	int i;
	if(n==0)
		return;
	for(i=1;i<=k;i++)
	{
		while(n%pri[i]==0)
		{
			n /= pri[i];
			cot[i] += m;
		}
	}
}
int main(void)
{
	int n, i, j, ok, sum, cnt, len;
	k = 0;
	for(i=2;i<=1000;i++)
	{
		if(p[i])
			continue;
		pri[++k] = i;
		for(j=i*i;j<=1000;j+=i)
			p[j] = 1;
	}
	while(scanf("%d", &n)!=EOF)
	{
		ok = 1;
		sum = cnt = 0;
		memset(cot, 0, sizeof(cot));
		for(i=1;i<=n;i++)
		{
			scanf("%d", &a[i]);
			if(a[i]==0 || a[i]>=n)
				ok = 0;
			if(a[i]!=-1)
			{
				cnt++;
				sum += a[i]-1;
				for(j=1;j<=a[i]-1;j++)
					Add(j, -1);
			}
		}
		if(ok==0 || n-2-sum<0)
			printf("0\n");
		else if(n==1)
		{
			if(a[1]<=0)  printf("1\n");
			else  printf("0\n");
		}
		else if(n==2)
		{
			if(a[1]>1 || a[2]>1 || a[1]==0 || a[2]==0)  printf("0\n");
			else  printf("1\n");
		}
		else
		{
			for(i=n-2-sum+1;i<=n-2;i++)
				Add(i, 1);
			Add(n-cnt, n-2-sum);
			memset(ans, 0, sizeof(ans));
			ans[1] = 1, len = 1;
			for(i=1;i<=k;i++)
			{
				while(cot[i])
				{
					cot[i]--;
					for(j=1;j<=len;j++)
						ans[j] *= pri[i];
					for(j=1;j<len;j++)
					{
						ans[j+1] += ans[j]/10;
						ans[j] %= 10;
					}
					while(ans[len]>=10)
					{
						ans[len+1] = ans[len]/10;
						ans[len++] %= 10;
					}
				}
			}
			for(i=len;i>=1;i--)
				printf("%d", ans[i]);
			printf("\n");
		}
	}
}