AT_abc331_g [ABC331G] Collect Them All

题目描述

箱子里有 $N$ 张卡片。每张卡片上写有一个 $1$ 到 $M$ 之间的整数,对于每个 $i=1,\ldots,M$,写有 $i$ 的卡片有 $C_i$ 张。 你手上有一本空白的笔记本,你会重复以下操作: - 从箱子中随机取出一张卡片。把卡片上写的整数记在笔记本上,然后把卡片放回箱子。 请你求出,直到笔记本上 $1$ 到 $M$ 的每个整数都至少被记过一次为止,所需操作次数的期望值,并对 $998244353$ 取模。 期望值 ${}\bmod{998244353}$ 的定义 本题中要求的期望值一定是有理数。在本题的约束下,设期望值为最简分数 $\frac{y}{x}$,保证 $x$ 不能被 $998244353$ 整除。此时,存在唯一的 $0\leq z

输入格式

输入通过标准输入给出,格式如下: > $N$ $M$ $C_1$ $\ldots$ $C_M$

输出格式

请输出答案。

说明/提示

### 限制条件 - $1\leq M\leq N\leq 2\times 10^5$ - $1\leq C_i$ - $\sum_{i=1}^{M}C_i=N$ - 输入均为整数 ### 样例解释 1 一系列操作可能如下进行: - 第一次取出的卡片上写着 $1$,笔记本上有 $1$ 一次。 - 第二次取出的卡片上写着 $1$,笔记本上有 $1$ 两次。 - 第三次取出的卡片上写着 $2$,笔记本上有 $1$ 两次,$2$ 一次。 所求的期望值为 $2\times\frac{1}{2}+3\times\frac{1}{4}+4\times\frac{1}{8}+\ldots=3$。 ### 样例解释 2 所求的期望值为 $\frac{21}{4}$,对 $998244353$ 取模后为 $748683270$。 ### 样例解释 3 所求的期望值为 $\frac{13943237577224054960759}{61980890084919934128}$。 由 ChatGPT 4.1 翻译