AT_abc412_f.[ABC412F] Socks 4
省选/NOI-
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
高桥君的抽屉里有 N 种颜色的袜子,第 i 种颜色的袜子有 Ai 只。
高桥君一开始在抽屉外还持有一只颜色为 C 的袜子。接下来,他会重复以下操作,直到满足结束条件为止:
- 从抽屉中等概率随机取出一只袜子。此时,如果抽屉外的两只袜子颜色相同,则操作结束。否则,从这两只袜子中选择一只放回抽屉。高桥君总是选择能使今后取袜子次数的期望值最小的那只袜子放回抽屉。
请计算操作结束前取袜子的次数的期望值,并对 998244353 取模输出。
在本题的约束下,期望值一定存在且为有理数。设其最简分数表示为 QP (Q>0),且 Q≡0(mod998244353)。因此,存在唯一的整数 R 满足 R×Q≡P(mod998244353), 0≤R<998244353。本题要求输出该 R 的值。
输入格式
输入为一行,格式如下:
N C A1 A2 … AN
输出格式
输出答案。
输入输出样例
输入#1
3 2 3 1 2
输出#1
249561092
输入#2
8 4 4 1 6 2 5 1 7 3
输出#2
393623786
说明/提示
限制条件
- 1≤N≤3×105
- 1≤C≤N
- 1≤Ai≤3000
- 输入的所有值均为整数
样例解释 1
取袜子的次数的期望值为 415。
由 ChatGPT 4.1 翻译
输入解题思路,AI测评打分。不知道怎么写?