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
Solution
感觉其实和树的计数那题差不多,时隔四年…HNOI为何对prufer编码情有独钟0 0
算是加强版吧
cnt为度数未定的点的个数,sum为prufer数列中已确定的个数
答案是
Csumn−2×sum!(d1−1)×(d2−1)×...×(di−1)×(n−2−sum)cnt
即:在n-2个位置里选sum个位置放已确定的数×sum个位置的排列×剩余的位置可以在度数未定的点里随便选
一开始忘记乘
Csumn−2
了…
还是要分解质因数(可以避免掉除法)
要用高精度,不过是高精乘低精,好评
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#define MAXN 1002
using namespace std;
int n,sum=0,tot=0,cnt=0;
int d[MAXN],pri[MAXN],num[MAXN];
bool jud[MAXN];
int ans[100000],len;
void getprime()
{
for(int i=2;i<=n;i++)
{
if(!jud[i])pri[++tot]=i;
for(int j=1;j<=tot&&i*pri[j]<=n;j++)
{
jud[i*pri[j]]=1;
if(i%pri[j]==0)break;
}
}
}
void add(int m,int n)
{
int i=1;
while(m>1)
{
while(m%pri[i]==0)
m/=pri[i],num[i]+=n;
i++;
}
}
void mult(int x)
{
int t=0;
for(int i=0,g=0;i<len||g!=0;i++)
{
g+=ans[t]*x;
ans[t]=g%10;
g/=10;
++t;
}
len=t;
}
void calc()
{
ans[0]=1,len=1;
for(int i=1;i<=tot;i++)
{
for(int j=1;j<=num[i];j++)
mult(pri[i]);
}
for(int i=1;i<=n-2-sum;i++)
mult(cnt);
}
void print()
{
for(int i=len-1;i>=0;i--)
printf("%d",ans[i]);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&d[i]);
if(d[i]==-1)
{
cnt++;
continue;
}
if(!d[i])
{
printf("0");
return 0;
}
sum+=d[i]-1;
}
if(sum>n-2)
{
printf("0");
return 0;
}
getprime();
for(int i=1;i<=sum;i++)add(i,1);
for(int i=1;i<=n;i++)
{
if(d[i]!=-1)
for(int j=1;j<=d[i]-1;j++)
add(j,-1);
}
for(int i=1;i<=n-2;i++)add(i,1);
for(int i=1;i<=sum;i++)add(i,-1);
for(int i=1;i<=n-2-sum;i++)add(i,-1);
calc();
print();
return 0;
}