AT_abc331_g.[ABC331G] Collect Them All
NOI/NOI+/CTSC
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
箱子里有 N 张卡片。每张卡片上写有一个 1 到 M 之间的整数,对于每个 i=1,…,M,写有 i 的卡片有 Ci 张。
你手上有一本空白的笔记本,你会重复以下操作:
- 从箱子中随机取出一张卡片。把卡片上写的整数记在笔记本上,然后把卡片放回箱子。
请你求出,直到笔记本上 1 到 M 的每个整数都至少被记过一次为止,所需操作次数的期望值,并对 998244353 取模。
期望值 mod998244353 的定义
本题中要求的期望值一定是有理数。在本题的约束下,设期望值为最简分数 xy,保证 x 不能被 998244353 整除。此时,存在唯一的 0≤z<998244353 满足 y≡xz(mod998244353),请输出 z。
输入格式
输入通过标准输入给出,格式如下:
N M C1 … CM
输出格式
请输出答案。
输入输出样例
输入#1
2 2 1 1
输出#1
3
输入#2
5 2 4 1
输出#2
748683270
输入#3
50 50 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
输出#3
244742906
输入#4
74070 15 1 2 3 11 22 33 111 222 333 1111 2222 3333 11111 22222 33333
输出#4
918012973
说明/提示
限制条件
- 1≤M≤N≤2×105
- 1≤Ci
- ∑i=1MCi=N
- 输入均为整数
样例解释 1
一系列操作可能如下进行:
- 第一次取出的卡片上写着 1,笔记本上有 1 一次。
- 第二次取出的卡片上写着 1,笔记本上有 1 两次。
- 第三次取出的卡片上写着 2,笔记本上有 1 两次,2 一次。
所求的期望值为 2×21+3×41+4×81+…=3。
样例解释 2
所求的期望值为 421,对 998244353 取模后为 748683270。
样例解释 3
所求的期望值为 6198089008491993412813943237577224054960759。
由 ChatGPT 4.1 翻译
输入解题思路,AI测评打分。不知道怎么写?