AT_abc331_g.[ABC331G] Collect Them All

NOI/NOI+/CTSC

通过率:0%

AC君温馨提醒

该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

箱子里有 NN 张卡片。每张卡片上写有一个 11 到 MM 之间的整数,对于每个 i=1,…,Mi=1,\ldots,M,写有 ii 的卡片有 CiC_i 张。

你手上有一本空白的笔记本,你会重复以下操作:

  • 从箱子中随机取出一张卡片。把卡片上写的整数记在笔记本上,然后把卡片放回箱子。

请你求出,直到笔记本上 11 到 MM 的每个整数都至少被记过一次为止,所需操作次数的期望值,并对 998244353998244353 取模。

期望值  mod 998244353{}\bmod{998244353} 的定义
本题中要求的期望值一定是有理数。在本题的约束下,设期望值为最简分数 yx\frac{y}{x},保证 xx 不能被 998244353998244353 整除。此时,存在唯一的 0≤z<9982443530\leq z<998244353 满足 y≡xz(mod998244353)y\equiv xz\pmod{998244353},请输出 zz。

输入格式

输入通过标准输入给出,格式如下:

NN MM C1C_1 …\ldots CMC_M

输出格式

请输出答案。

输入输出样例

  • 输入#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×1051\leq M\leq N\leq 2\times 10^5
  • 1≤Ci1\leq C_i
  • ∑i=1MCi=N\sum_{i=1}^{M}C_i=N
  • 输入均为整数

样例解释 1

一系列操作可能如下进行:

  • 第一次取出的卡片上写着 11,笔记本上有 11 一次。
  • 第二次取出的卡片上写着 11,笔记本上有 11 两次。
  • 第三次取出的卡片上写着 22,笔记本上有 11 两次,22 一次。
    所求的期望值为 2×12+3×14+4×18+…=32\times\frac{1}{2}+3\times\frac{1}{4}+4\times\frac{1}{8}+\ldots=3。

样例解释 2

所求的期望值为 214\frac{21}{4},对 998244353998244353 取模后为 748683270748683270。

样例解释 3

所求的期望值为 1394323757722405496075961980890084919934128\frac{13943237577224054960759}{61980890084919934128}。

由 ChatGPT 4.1 翻译

输入解题思路,AI测评打分。不知道怎么写?

首页