AT_abc326_e.[ABC326E] Revenge of "The Salary of AtCoder Inc."
普及+/提高
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
青木是 AtCoder 公司的员工,他本月的工资通过一个整数 N 和一个长度为 N 的数列 A 按如下方式决定。
首先,青木会得到一个有 N 个面的骰子(每个面上分别标有 1 到 N 的整数,掷出每个面的概率相等),以及一个变量 x=0。
然后,重复以下步骤直到结束:
- 掷一次骰子,掷出的点数记为 y。
- 如果 x<y,则发放 Ay 日元,并将 x 更新为 y。
- 否则,流程结束。
青木本月的工资为通过上述流程累计发放的金额之和。
请计算青木本月工资的期望值,并对 998244353 取模后输出。
期望值 mod998244353 的定义
本题中要求的期望值一定是有理数。在本题的约束下,设期望值为最简分数 xy,保证 x 不会被 998244353 整除。此时,存在唯一的 0≤z<998244353 满足 y≡xz(mod998244353),请输出 z。
输入格式
输入通过标准输入给出,格式如下:
N A1 A2 … AN
输出格式
请输出答案。
输入输出样例
输入#1
3 3 2 6
输出#1
776412280
输入#2
1 998244352
输出#2
998244352
输入#3
9 3 14 159 2653 58979 323846 2643383 27950288 419716939
输出#3
545252774
说明/提示
数据范围
- 输入均为整数。
- 1≤N≤3×105
- 0≤Ai<998244353
样例解释 1
一种流程如下:
- 初始时,x=0。
- 掷一次骰子,结果为 1。由于 0<1,发放 A1=3 日元,x 更新为 1。
- 再掷一次骰子,结果为 3。由于 1<3,发放 A3=6 日元,x 更新为 3。
- 再掷一次骰子,结果为 1。由于 3≥1,流程结束。
在这个例子中,青木本月的工资为 9 日元。
本月工资的期望值可以计算为 949 日元,模 998244353 意义下为 776412280。
由 ChatGPT 4.1 翻译
输入解题思路,AI测评打分。不知道怎么写?