【BZOJ1005】[HNOI2008]明明的烦恼（prufer序列）

【BZOJ1005】[HNOI2008]明明的烦恼（prufer序列）

题面

BZOJ
洛谷

题解

戳这里

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define ll long long
#define MAX 1010
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
const int yw=10000;
struct BigNum
{
    ll s[MAX*2];int ws;
    void output()
        {
            printf("%lld",s[ws]);
            for(int i=ws-1;i;--i)
                printf("%04lld",s[i]);
            puts("");
        }
    void clear(){memset(s,0,sizeof(s));ws=0;}
}ans;
BigNum operator*(BigNum a,int b)
{
    int ws=a.ws;BigNum ret;ret.clear();
    for(int i=1;i<=ws;++i)ret.s[i]=a.s[i]*b;
    for(int i=1;i<=ws;++i)ret.s[i+1]+=ret.s[i]/yw,ret.s[i]%=yw;
    while(ret.s[ws+1])++ws,ret.s[ws+1]+=ret.s[ws]/yw,ret.s[ws]%=yw;
    ret.ws=ws;return ret;
}
int sum,a[MAX],cnt,n;
int p1[MAX],p2[MAX];
void add(int *p,int x)
{
    for(int i=2;i*i<=x;++i)
        while(x%i==0)x/=i,++p[i];
    if(x>1)++p[x];
}
int main()
{
    n=read();
    for(int i=1;i<=n;++i)
    {
        a[i]=read();if(a[i]==-1)continue;
        ++cnt;sum+=a[i]-1;
    }
    if(sum+n>2*(n-1)){puts("0");return 0;}
    for(int i=n-2;i;--i)add(p1,i);
    for(int i=n-2-sum;i;--i)add(p2,i);
    for(int i=1;i<=n;++i)
        if(a[i]!=-1)
            for(int j=1;j<a[i];++j)
                add(p2,j);
    for(int i=1;i<=n-2-sum;++i)add(p1,n-cnt);
    for(int i=1;i<=n;++i)p1[i]-=p2[i];
    ans.s[1]=1;ans.ws=1;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=p1[i];++j)
            ans=ans*i;
    ans.output();
    return 0;           
}