【BZOJ1005】明明的烦恼

【不看题解一辈子想不出来系列之一】

树的prufer编码，具体内容见黄学长博客。。。
http://hzwer.com/3272.html
实现小技巧就是开全局数组记答案各质因数的次数最后统一乘，只有bign*int 所以可以压位

<del>然而数据规模太小所以我什么优化都没有加

#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define maxn 1005
void _read(int &x)
{
x=0; char ch=getchar(); bool flag=false;
while(ch<'0' || ch>'9'){if(ch=='-')flag=1; ch=getchar();}
while(ch>='0' && ch<='9'){x=x*10+ch-'0'; ch=getchar();}
if(flag)x=-x;
return ;
}
int n,cnt,sum,d[maxn],pre_cnt,pre[2*maxn],a[2*maxn],b[2*maxn];
bool v[2*maxn],wrong;
void getpre(int T)//分解质因数的准备工作
{
memset(v,0,sizeof(v));
for(int i=2;i<=T;i++)if(!v[i])
{
pre[++pre_cnt]=i;
for(int j=i;j*i<=T;j++)v[j*i]=1;
}
return ;
}
void Init()
{
wrong=false;
_read(n);
sum=0;cnt=0;
for(int i=1;i<=n;i++)_read(d[i]);

for(int i=1;i<=n;i++)if(d[i]==0){wrong=true;return;}
for(int i=1;i<=n;i++)if(d[i]!=-1){cnt++,sum+=(d[i]-1);}
getpre(2000);
return ;
}
void fj(int now)//将某个数的阶乘唯一分解
{
int t=0;
memset(b,0,sizeof(b));
if(now==1)return ;
for(int i=1;i<=pre_cnt;i++)
{
t=pre[i];
while(t<=now){b[i]+=now/t; t=t*pre[i];}
}
return ;
}
void getC(int x,int y)
{
fj(y);for(int i=1;i<=pre_cnt;i++)a[i]+=b[i];
fj(x);for(int i=1;i<=pre_cnt;i++)a[i]-=b[i];
fj(y-x);for(int i=1;i<=pre_cnt;i++)a[i]-=b[i];
return ;
}
void getA()
{
fj(sum);for(int i=1;i<=pre_cnt;i++)a[i]+=b[i];
for(int j=1;j<=n;j++)if(d[j]-1>0)
{
fj(d[j]-1);
for(int i=1;i<=pre_cnt;i++)a[i]-=b[i];
}
return ;
}
void getPower()
{
memset(b,0,sizeof(b));
if(cnt<n)
{
int t=n-cnt;
for(int i=1;i<=pre_cnt;i++)
{
if(t<=1)break;
while(t%pre[i]==0 && t>0){b[i]++; t=t/pre[i];}
}
for(int i=1;i<=pre_cnt;i++)b[i]*=(n-2-sum);
}
for(int i=1;i<=pre_cnt;i++)a[i]+=b[i];
return ;
}
struct bign
{
int a[4005],n;
bign()
{
memset(a,0,sizeof(a)); n=1; a[1]=1;
}
friend bign operator *(bign x,int y)
{
bign b;
b.n=x.n+5;
for(int i=1;i<=b.n;i++)b.a[i]=x.a[i]*y;
for(int i=1;i<=b.n;i++)if(b.a[i]>=10){b.a[i+1]+=b.a[i]/10; b.a[i]=b.a[i]%10;}
while(b.a[b.n]==0)b.n--;
return b;
}
}ans;
void Output()
{

for(int i=1;i<=pre_cnt;i++)
{
for(int j=1;j<=a[i];j++)ans=ans*pre[i];
}
for(int i=ans.n;i>=1;i--)putchar(ans.a[i]+'0');
putchar('\n');
return ;
}
void work()
{
memset(a,0,sizeof(a));//a数组记录答案中各质因数的次数
//计算组合数
getC(sum,n-2);
//计算可重集排列
getA();
//计算无限制项
getPower();
Output();
return ;
}
int main()
{
freopen("in.txt","r",stdin);
Init();
if(n==1)
{
if(d[1]==0)printf("1\n");
else printf("0\n");
return 0;
}
if(sum<=n-2 && wrong==false)work();
else printf("0\n");
return 0;
}