直接套用 prufer 序列公式即可,但是要特判不合法的情况.
code:
#include <cstdio> #include <algorithm> #define ll long long #define N 200 #define setIO(s) freopen(s".in","r",stdin) using namespace std; int bu[N]; void update(int x,int v) { for(int i=2;i*i<=x;++i) for(;x%i==0;x/=i) bu[i]+=v; if(x>1) bu[x]+=v; } int main() { // setIO("input"); int i,j,n,s=0; scanf("%d",&n); for(i=2;i<=n-2;++i) update(i,1); for(i=1;i<=n;++i) { int x; scanf("%d",&x),s+=x; if(x==0&&n!=1) { printf("0\n"); return 0; } for(j=2;j<x;++j) update(j,-1); } if(s!=2*(n-1)) printf("0\n"); else { ll ans=1; for(i=1;i<N;++i) { for(j=1;j<=bu[i];++j) ans*=i; } printf("%lld\n",ans); } return 0; }