AT_arc156_b [ARC156B] Mex on Blackboard

题目描述

对于由有限个非负整数组成的多重集 $S$,定义 $\mathrm{mex}(S)$ 为不属于 $S$ 的最小非负整数。例如,$\mathrm{mex}(\lbrace 0,0,1,3\rbrace )=2$,$\mathrm{mex}(\lbrace 1 \rbrace)=0$,$\mathrm{mex}(\lbrace \rbrace)=0$。 黑板上写有 $N$ 个非负整数,第 $i$ 个数为 $A_i$。 你需要恰好进行 $K$ 次如下操作: - 从黑板上的非负整数中选出 $0$ 个或多个,组成多重集 $S$,然后将 $\mathrm{mex}(S)$ 写到黑板上 $1$ 次。 请你求出,经过 $K$ 次操作后,黑板上可能出现的非负整数多重集的种数,结果对 $998244353$ 取模。

输入格式

输入以以下格式从标准输入给出。 > $N$ $K$ $A_1$ $A_2$ $\ldots$ $A_N$

输出格式

请输出答案。

说明/提示

## 限制条件 - $1 \leq N,K \leq 2\times 10^5$ - $0 \leq A_i \leq 2\times 10^5$ - 输入的所有数均为整数 ## 样例解释 1 操作后可能得到的多重集有以下 $3$ 种: - $\lbrace 0,0,1,3 \rbrace$ - $\lbrace 0,1,1,3 \rbrace$ - $\lbrace 0,1,2,3 \rbrace$ 例如,$\lbrace 0,1,1,3 \rbrace$ 可以通过选择黑板上的 $0$,令 $S=\lbrace 0\rbrace$,然后进行操作得到。 ## 样例解释 2 操作后可能得到的多重集有以下 $2$ 种: - $\lbrace 0,0,0 \rbrace$ - $\lbrace 0,0,1 \rbrace$ 注意,操作时可以选择 $0$ 个整数。 由 ChatGPT 4.1 翻译