atcoder#NIKKEI20192QUALB. Counting of Trees

Counting of Trees

题目描述

N N 要素からなる整数列 D1,...,DN D_1,...,D_N が与えられます。頂点に 1 1 から N N の番号が付けられた N N 頂点からなる木であって、 以下の条件をみたすものの個数を 998244353 998244353 で割ったあまりを求めてください。

  • 1 1 以上 N N 以下の任意の整数 i i に対して、頂点 1 1 と頂点 i i の距離が Di D_i である。

输入格式

入力は以下の形式で標準入力から与えられる。

N N D1 D_1 D2 D_2 ... ... DN D_N

输出格式

答えを出力せよ。

题目大意

  • 一棵有 nn 个节点,以 11 为根的有根树是好的,当且仅当第 ii 个节点的深度为 aia_i(根的深度为 00)。
  • 计算有多少好的树,答案对 998244353998244353 取模。
  • 1n105,0ai<n1\leq n\leq10^5,0\leq a_i<n
4
0 1 1 2
2
4
1 1 1 1
0
7
0 3 2 1 2 2 1
24

提示

注記

  • N N 頂点の木とは N N 頂点 N1 N-1 辺からなる連結無向グラフのことであり、2 2 頂点の距離とは一方から他方への最短路に用いられる辺の個数を指します。
  • 2 2 つの木が異なるとは、ある 2 2 頂点 x x , y y が存在して、x x y y の間に一方の木では辺が存在し、 もう一方の木では辺が存在しないことを指します。

制約

  • 1  N  105 1\ ≦\ N\ ≦\ 10^5
  • 0  Di  N1 0\ ≦\ D_i\ ≦\ N-1

Sample Explanation 1

例えば、(1,2) (1,2) , (1,3) (1,3) , (2,4) (2,4) の間に辺があるような木が条件をみたします。