AT_abc352_g [ABC352G] Socks 3
题目描述
高桥君的抽屉里有各种颜色的袜子。袜子的颜色用 $1$ 到 $N$ 之间的整数表示,颜色 $i$ 的袜子有 $A_i\ (\geq\ 2)$ 双。
高桥君打算通过以下操作选择今天要穿的袜子:
- 不断从抽屉中随机且等概率地取出一只袜子,直到在已经取出的袜子中,某种颜色的袜子可以组成一对为止。取出的袜子不会放回抽屉。
请你求出高桥君从抽屉中取袜子的次数的期望值,并对 $998244353$ 取模。
所谓对 $998244353$ 取模的期望值,是指期望值一定可以表示为一个最简分数 $\frac{y}{x}$,并且在本题的约束下,$x$ 保证不会被 $998244353$ 整除。此时,存在唯一的 $0$ 到 $998244352$ 之间的整数 $z$ 满足 $xz \equiv y \pmod{998244353}$,请输出这个 $z$。
输入格式
输入为以下格式,从标准输入读入。
> $N$ $A_1$ $A_2$ $\dots$ $A_N$
输出格式
请输出答案。
说明/提示
## 约束条件
- $1 \leq N \leq 3 \times 10^5$
- $2 \leq A_i \leq 3000$
- 输入均为整数
## 样例解释 1
例如,可以按如下方式进行操作:
1. 从抽屉中取出一只颜色为 $1$ 的袜子。此时抽屉中还剩 $1$ 只颜色为 $1$ 的袜子和 $2$ 只颜色为 $2$ 的袜子。
2. 从抽屉中取出一只颜色为 $2$ 的袜子。此时抽屉中还剩 $1$ 只颜色为 $1$ 的袜子和 $1$ 只颜色为 $2$ 的袜子。
3. 从抽屉中取出一只颜色为 $1$ 的袜子。此时已经取出的袜子中有 $2$ 只颜色为 $1$ 的袜子和 $1$ 只颜色为 $2$ 的袜子,可以组成一对颜色为 $1$ 的袜子,操作结束。
在这个例子中,高桥君取袜子的次数为 $3$ 次。
高桥君取袜子的次数有 $\frac{2}{3}$ 的概率为 $3$ 次,有 $\frac{1}{3}$ 的概率为 $2$ 次,所以期望值为 $3 \times \frac{2}{3} + 2 \times \frac{1}{3} = \frac{8}{3} \equiv 665496238 \pmod{998244353}$。
由 ChatGPT 4.1 翻译