atcoder#ABC303H. [ABC303Ex] Constrained Tree Degree

[ABC303Ex] Constrained Tree Degree

题目描述

整数 N N 及び 1 1 以上 N1 N-1 以下の整数からなる集合 S={ S1,S2,,SK} S=\lbrace\ S_1,S_2,\ldots,S_K\rbrace が与えられます。

頂点に 1 1 から N N の番号がついた N N 頂点の木 T T のうち、以下の条件を満たすものの個数を 998244353 998244353 で割った余りを答えてください。

  • 任意の i (1 i  N) i\ (1\leq\ i\ \leq\ N) について、T T の頂点 i i の次数を di d_i としたとき、 di S d_i\in\ S

输入格式

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

N N K K S1 S_1 \ldots SK S_K

输出格式

条件を満たす木 T T の個数を 998244353 998244353 で割った余りを出力せよ。

题目大意

给定一个长度为 KK 的正整数序列 SS,求有多少个不同的树 TT 使得:

  • TT 中有 NN 个节点。

  • 对于 TT 中的任意一个节点 ii 的度数 did_i,有 diSd_i\in S

4 2
1 3
4
10 5
1 2 3 5 6
68521950
100 5
1 2 3 14 15
888770956

提示

制約

  • 2 N  2× 105 2\leq\ N\ \leq\ 2\times\ 10^5
  • 1 K  N1 1\leq\ K\ \leq\ N-1
  • 1 S1 < S2 <  < SK  N1 1\leq\ S_1\ <\ S_2\ <\ \ldots\ <\ S_K\ \leq\ N-1
  • 入力は全て整数

Sample Explanation 1

ある 1 1 つの頂点の次数が 3 3 であり、ほかの頂点の次数が 1 1 であるような木が条件を満たします。よって答えは 4 4 個です。

Sample Explanation 3

個数を 998244353 998244353 で割った余りを出力してください。