AT_abc303_h [ABC303Ex] Constrained Tree Degree

题目描述

给定一个整数 $N$,以及一个由 $1$ 到 $N-1$ 之间的整数构成的集合 $S=\lbrace S_1, S_2, \ldots, S_K \rbrace$。 请计算满足以下条件的 $N$ 个顶点的树 $T$(顶点编号为 $1$ 到 $N$)的个数,并输出其对 $998244353$ 取模的结果。 - 对于任意 $i\ (1\leq i \leq N)$,设 $T$ 中顶点 $i$ 的度数为 $d_i$,则 $d_i \in S$。

输入格式

输入以如下格式从标准输入读入。 > $N$ $K$ $S_1$ $S_2$ $\ldots$ $S_K$

输出格式

输出满足条件的树 $T$ 的个数对 $998244353$ 取模的结果。

说明/提示

## 限制条件 - $2 \leq N \leq 2 \times 10^5$ - $1 \leq K \leq N-1$ - $1 \leq S_1 < S_2 < \ldots < S_K \leq N-1$ - 所有输入均为整数 ## 样例解释 1 存在一种情况,其中一个顶点的度数为 $3$,其余顶点的度数均为 $1$,这样的树满足条件。因此答案为 $4$ 个。 ## 样例解释 3 请输出个数对 $998244353$ 取模的结果。 由 ChatGPT 4.1 翻译