[luogu2624 HNOI2008]明明的烦恼 (prufer+高精)

传送门

Solution

根据prufer序列做的题，具体可以看这里
还知道了一种避免高精除的方法quq

Code

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define F(i,a,b) for(register int i=(a);i<=(b);i++)
#define R(i,a,b) for(register int i=(b);i>=(a);i--)
using namespace std;

inline int read() {
    int x=0,f=1;char c=getchar();
    while(!isdigit(c)) {if(c=='-')f=-f;c=getchar();}
    while(isdigit(c)) x=(x<<1)+(x<<3)+c-48,c=getchar();
    return x*f;
}

const int N=1010,D=10000;

struct Bign{
    int da[N<<2],wei;
    Bign() {clear();}
    void out() {
        printf("%d",da[wei]);
        R(i,1,wei-1) printf("%04d",da[i]);
        putchar('\n');
    }
    void clear() {memset(da,0,sizeof(da));wei=0;}
}ans;
Bign operator*(Bign a,int b) {
    Bign res; res.wei=a.wei; int &wei=res.wei;
    F(i,1,wei) res.da[i]=a.da[i]*b;
    F(i,1,wei) res.da[i+1]+=res.da[i]/D,res.da[i]%=D;
    while(res.da[wei+1]) wei++,res.da[wei+1]=res.da[wei]/D,res.da[wei]%=D;
    return res;
}

int n,sum,cnt;
int a[N],p1[N],p2[N];

void get_p(int *p,int x) {
    int sqr=sqrt(x);
    F(i,2,sqr) while(x%i==0) x/=i,p[i]++;
    if(x>1) p[x]++;
}

int main() {
    n=read();
    F(i,1,n) {
        a[i]=read();if(a[i]==-1) continue;
        cnt++;sum+=a[i]-1;
        F(j,1,a[i]-1) get_p(p2,j);
    }
    if(sum>n-2) return putchar('0'),0;
    F(i,1,n-2) get_p(p1,i); 
    F(i,1,n-2-sum) get_p(p2,i),get_p(p1,n-cnt);
    F(i,1,n) p1[i]-=p2[i];
    ans.da[1]=1;ans.wei=1;
    F(i,1,n) F(j,1,p1[i]) ans=ans*i;
    ans.out();
    return 0;
}