Counting of Trees
题意翻译
* 一棵有 $n$ 个节点,以 $1$ 为根的有根树是好的,当且仅当第 $i$ 个节点的深度为 $a_i$(根的深度为 $0$)。
* 计算有多少好的树,答案对 $998244353$ 取模。
* $1\leq n\leq10^5,0\leq a_i<n$
题目描述
[problemUrl]: https://atcoder.jp/contests/nikkei2019-2-qual/tasks/nikkei2019_2_qual_b
$ N $ 要素からなる整数列 $ D_1,...,D_N $ が与えられます。頂点に $ 1 $ から $ N $ の番号が付けられた $ N $ 頂点からなる木であって、 以下の条件をみたすものの個数を $ 998244353 $ で割ったあまりを求めてください。
- $ 1 $ 以上 $ N $ 以下の任意の整数 $ i $ に対して、頂点 $ 1 $ と頂点 $ i $ の距離が $ D_i $ である。
输入输出格式
输入格式
入力は以下の形式で標準入力から与えられる。
> $ N $ $ D_1 $ $ D_2 $ $ ... $ $ D_N $
输出格式
答えを出力せよ。
输入输出样例
输入样例 #1
4
0 1 1 2
输出样例 #1
2
输入样例 #2
4
1 1 1 1
输出样例 #2
0
输入样例 #3
7
0 3 2 1 2 2 1
输出样例 #3
24
说明
### 注記
- $ N $ 頂点の木とは $ N $ 頂点 $ N-1 $ 辺からなる連結無向グラフのことであり、$ 2 $ 頂点の距離とは一方から他方への最短路に用いられる辺の個数を指します。
- $ 2 $ つの木が異なるとは、ある $ 2 $ 頂点 $ x $, $ y $ が存在して、$ x $ と $ y $ の間に一方の木では辺が存在し、 もう一方の木では辺が存在しないことを指します。
### 制約
- $ 1\ ≦\ N\ ≦\ 10^5 $
- $ 0\ ≦\ D_i\ ≦\ N-1 $
### Sample Explanation 1
例えば、$ (1,2) $, $ (1,3) $, $ (2,4) $ の間に辺があるような木が条件をみたします。