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
Solution
介绍一下purfer序列,一个编号1~n的树与一个长度为n-2的purfer序列一一对应,下面看看两者如何一一对应
对于一棵1~n的树,用n-2步得到与其对应的purfer序列,第i步找到当前度数为1且编号最小的点u,将与其相连的点v的编号作为purfer序列的第i个元素,然后将u删去,做完n-2步我们得到一个长度为n-2的序列和一个只有两个节点的树,通过以上步骤可知一棵树只会得到一个唯一的purfer序列
对于长度为n-2的purfer序列,首先将1~n这n个点的度数初始化为1,然后再加上这个点的编号在purfer序列中出现的次数得到每个点的度数,之后做n-2步得到一棵树,第i步找到度数为1且编号最小的点u,然后取出序列第i个元素v,在树上构造一条u-v的边,然后将u,v的度数减一,重复以上过程得到一棵n个节点的树,通过以上步骤可知一个purfer序列对应一棵唯一的树
知道以上知识后我们再来看这道题,求合法的树的个数其实就变成求合法的purfer序列的个数,对于度数有限制的点i,假设有cnt个,我们知道其在purfer序列中出现的次数就是其度数a[i]减一,令,那么度数无限制的点在purfer序列中出现的次数之和为n-2-sum,每次可以选择剩余n-cnt个点中的任何一个点,故满足条件的purfer序列有个,当然这是在合法且n>2的情况下的计算方法,n为1和2时特判,不合法情况就是某个点度数为0或者大于n-1,还有sum>n-2的情况,对于合法情况,上面的结果也非常大,故要使用高精度乘法,为避免除法的出现,对于阶乘和组合数可以素因子分解后就变成若干素数的乘积了
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define mod 10000
#define maxn 1111
int ans[111111],len;
void mul(int n)
{
int k=0;
for(int i=0;i<len;i++)
{
int temp=ans[i]*n+k;
ans[i]=temp%mod;
k=temp/mod;
}
if(k!=0)ans[len]=k,len++;
}
void output()
{
printf("%d",ans[len-1]);
for(int i=len-2;i>=0;i--)
printf("%04d",ans[i]);
printf("\n");
}
int prime[maxn],res,mark[maxn];
void get_prime()
{
memset(mark,0,sizeof(mark));
res=0;
for(int i=2;i<maxn;i++)
if(!mark[i])
{
prime[res++]=i;
for(int j=i;j<maxn;j+=i)mark[j]=1;
}
}
int n,a[maxn],sum,cnt,num[maxn];
void deal(int n,int sta)
{
for(int i=0;i<res&&prime[i]<=n;i++)
{
int m=n;
while(m)
{
num[i]+=sta*(m/prime[i]);
m/=prime[i];
}
}
}
int main()
{
get_prime();
scanf("%d",&n);
int gg=0;
memset(num,0,sizeof(num));
cnt=sum=0;
len=1,ans[0]=1;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]==0||a[i]>=n)gg=1;
if(a[i]==-1)continue;
cnt++,sum+=a[i]-1;
}
if(n==1)
{
if(a[1]==-1||a[1]==0)printf("1\n");
else printf("0\n");
return 0;
}
if(n==2)
{
if((a[1]==-1||a[1]==1)&&(a[2]==-1||a[2]==1))printf("1\n");
else printf("0\n");
return 0;
}
if(gg||sum>n-2)
{
printf("0\n");
return 0;
}
deal(n-2,1),deal(n-2-sum,-1);
for(int i=1;i<=n;i++)
if(a[i]!=-1)deal(a[i]-1,-1);
for(int i=0;i<res;i++)
while(num[i]--)mul(prime[i]);
for(int i=1;i<=n-2-sum;i++)mul(n-cnt);
output();
return 0;
}