AT_abc352_g.[ABC352G] Socks 3
省选/NOI-
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
高桥君的抽屉里有各种颜色的袜子。袜子的颜色用 1 到 N 之间的整数表示,颜色 i 的袜子有 Ai (≥ 2) 双。
高桥君打算通过以下操作选择今天要穿的袜子:
- 不断从抽屉中随机且等概率地取出一只袜子,直到在已经取出的袜子中,某种颜色的袜子可以组成一对为止。取出的袜子不会放回抽屉。
请你求出高桥君从抽屉中取袜子的次数的期望值,并对 998244353 取模。
所谓对 998244353 取模的期望值,是指期望值一定可以表示为一个最简分数 xy,并且在本题的约束下,x 保证不会被 998244353 整除。此时,存在唯一的 0 到 998244352 之间的整数 z 满足 xz≡y(mod998244353),请输出这个 z。
输入格式
输入为以下格式,从标准输入读入。
N A1 A2 … AN
输出格式
请输出答案。
输入输出样例
输入#1
2 2 2
输出#1
665496238
输入#2
1 352
输出#2
2
输入#3
6 1796 905 2768 253 2713 1448
输出#3
887165507
说明/提示
约束条件
- 1≤N≤3×105
- 2≤Ai≤3000
- 输入均为整数
样例解释 1
例如,可以按如下方式进行操作:
- 从抽屉中取出一只颜色为 1 的袜子。此时抽屉中还剩 1 只颜色为 1 的袜子和 2 只颜色为 2 的袜子。
- 从抽屉中取出一只颜色为 2 的袜子。此时抽屉中还剩 1 只颜色为 1 的袜子和 1 只颜色为 2 的袜子。
- 从抽屉中取出一只颜色为 1 的袜子。此时已经取出的袜子中有 2 只颜色为 1 的袜子和 1 只颜色为 2 的袜子,可以组成一对颜色为 1 的袜子,操作结束。
在这个例子中,高桥君取袜子的次数为 3 次。
高桥君取袜子的次数有 32 的概率为 3 次,有 31 的概率为 2 次,所以期望值为 3×32+2×31=38≡665496238(mod998244353)。
由 ChatGPT 4.1 翻译
输入解题思路,AI测评打分。不知道怎么写?