Counting of Trees

传送门：https://atcoder.jp/contests/nikkei2019-2-qual/tasks/nikkei2019_2_qual_b

题意

  有一颗 n 个节点的树，已知的是这 n 个节点分别到 根节点（节点1）的距离，问你这棵树的同分异构体的数目。
  Sample1 为例：两种：(1,2)，(1,3)，(2,4) 或者 (1,2)，(1,3)，(3,4)

思路

1. 首先，节点 1 到自己本身的距离是 0，也就是 D₁ = 1，反之输出 0；然后只有一颗树，所以的话，距离是 0 的D_i 只能有一个，反之输出 0；还有就是他的距离是连续出现的，不会出现间断，列如：0 1 1 3 5 这种情况的结果是 0；
2. 有排列组合的性质，我们知道是乘法关系，即当前层及以上层数的总种类 = 上一层及以上层的总种类 * 当前层的总种类。当前层的总次数 = 上一层的节点数^{这一层的节点数};

#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<cctype>
#include<vector>
#include<stack>
#include<queue>
#include<ctime>
#include<map>
#define ll long long
#define ld long double
#define ull unsigned long long
using namespace std;
const ll mod = 998244353;
const int maxn = 101001;
ll a[maxn];
ll mul(ll x,ll y,ll z)
{
    ll sm = (ld)x/z*y;
    return ((ull)x*y-(ull)sm*z+z)%z;
}
ll qpow(ll x,ll y,ll z)
{
    ll ans = 1;
    x %= z;
    for(ll i=y;i;i>>=1){
        if(i&1) ans = mul(ans,x,z);
        x = mul(x,x,z);
    }
    return ans%z;
}
int main(void)
{
    int n,x;
    scanf("%d",&n);
    cin>>x;	//判断首位是否为0
    if(x!=0){
        printf("0\n");
        return 0;
    }
    a[x]++;
    for(int i=1;i<n;i++){
        scanf("%d",&x);
        a[x]++;
    }
    if(a[0]>1){	//判断是否多个0
        printf("0\n");
        return 0;
    }
    int ret = 0;
    ll ans = 1,pr = 1;
    for(int i=0;i<n;i++){
        if(a[i]==0){	//判断是否有间隔
            if(ret!=n)  ans = 0;
            break;
        }
        ans = mul(ans,qpow(pr,a[i],mod),mod);
        pr = a[i];
        ret += (int)a[i];
    }
    printf("%lld\n",ans);
    return 0;
}