AT_abc231_g.[ABC231G] Balls in Boxes

NOI/NOI+/CTSC

通过率:0%

AC君温馨提醒

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

题目描述

有 NN 个编号为 11 到 NN 的箱子。最初,第 ii 个箱子中有 AiA_i 个球。

你需要重复进行 KK 次如下操作:

  • 从 NN 个箱子中等概率地随机选择一个(每次操作的选择相互独立)。在选中的箱子中加入 11 个球。

KK 次操作结束后,设第 ii 个箱子中的球数为 BiB_i,则得分为所有箱子球数的乘积 ∏i=1NBi\prod_{i=1}^{N} B_i。

请你计算得分的期望值,并对 998244353998244353 取模。

输入格式

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

NN KK A1A_1 A2A_2 …\ldots ANA_N

输出格式

输出答案。

输入输出样例

  • 输入#1

    3 1
    1 2 3

    输出#1

    665496245
  • 输入#2

    2 2
    1 2

    输出#2

    499122182
  • 输入#3

    10 1000000000
    998244350 998244351 998244352 998244353 998244354 998244355 998244356 998244357 998244358 998244359

    输出#3

    138512322

说明/提示

注意

若所求期望值可表示为最简分数 p/qp/q,则满足 rq≡p(mod998244353)rq\equiv p\pmod{998244353} 且 0≤r<9982443530\leq r<998244353 的整数 rr 在本题的约束下是唯一确定的。这个 rr 就是你需要输出的答案。

约束条件

  • 1≤N≤10001\leq N\leq 1000
  • 1≤K≤1091\leq K\leq 10^9
  • 0≤Ai≤1090\leq A_i\leq 10^9

样例解释 1

操作后,得分如下:

  • 操作选择箱子 11 时,2×2×3=122\times 2\times 3=12
  • 操作选择箱子 22 时,1×3×3=91\times 3\times 3=9
  • 操作选择箱子 33 时,1×2×4=81\times 2\times 4=8

因此,期望值为 13(12+9+8)=293\frac{1}{3}(12+9+8)=\frac{29}{3}。模 998244353998244353 后为 665496245665496245。

样例解释 2

操作后,得分如下:

  • 第 11 次操作选箱 11,第 22 次操作选箱 11:3×2=63\times 2=6
  • 第 11 次操作选箱 11,第 22 次操作选箱 22:2×3=62\times 3=6
  • 第 11 次操作选箱 22,第 22 次操作选箱 11:2×3=62\times 3=6
  • 第 11 次操作选箱 22,第 22 次操作选箱 22:1×4=41\times 4=4

因此,期望值为 14(6+6+6+4)=112\frac{1}{4}(6+6+6+4)=\frac{11}{2}。

由 ChatGPT 4.1 翻译

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

首页