AT_abc231_g.[ABC231G] Balls in Boxes
NOI/NOI+/CTSC
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
有 N 个编号为 1 到 N 的箱子。最初,第 i 个箱子中有 Ai 个球。
你需要重复进行 K 次如下操作:
- 从 N 个箱子中等概率地随机选择一个(每次操作的选择相互独立)。在选中的箱子中加入 1 个球。
K 次操作结束后,设第 i 个箱子中的球数为 Bi,则得分为所有箱子球数的乘积 ∏i=1NBi。
请你计算得分的期望值,并对 998244353 取模。
输入格式
输入通过标准输入给出,格式如下:
N K A1 A2 … AN
输出格式
输出答案。
输入输出样例
输入#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/q,则满足 rq≡p(mod998244353) 且 0≤r<998244353 的整数 r 在本题的约束下是唯一确定的。这个 r 就是你需要输出的答案。
约束条件
- 1≤N≤1000
- 1≤K≤109
- 0≤Ai≤109
样例解释 1
操作后,得分如下:
- 操作选择箱子 1 时,2×2×3=12
- 操作选择箱子 2 时,1×3×3=9
- 操作选择箱子 3 时,1×2×4=8
因此,期望值为 31(12+9+8)=329。模 998244353 后为 665496245。
样例解释 2
操作后,得分如下:
- 第 1 次操作选箱 1,第 2 次操作选箱 1:3×2=6
- 第 1 次操作选箱 1,第 2 次操作选箱 2:2×3=6
- 第 1 次操作选箱 2,第 2 次操作选箱 1:2×3=6
- 第 1 次操作选箱 2,第 2 次操作选箱 2:1×4=4
因此,期望值为 41(6+6+6+4)=211。
由 ChatGPT 4.1 翻译
输入解题思路,AI测评打分。不知道怎么写?