传送门: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 到自己本身的距离是 0,也就是 D1 = 1,反之输出 0;然后只有一颗树,所以的话,距离是 0 的Di 只能有一个,反之输出 0;还有就是他的距离是连续出现的,不会出现间断,列如:0 1 1 3 5 这种情况的结果是 0;
- 有排列组合的性质,我们知道是乘法关系,即当前层及以上层数的总种类 = 上一层及以上层的总种类 * 当前层的总种类。当前层的总次数 = 上一层的节点数这一层的节点数;
#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;
}