AT_abc326_e [ABC326E] Revenge of "The Salary of AtCoder Inc."
题目描述
青木是 AtCoder 公司的员工,他本月的工资通过一个整数 $N$ 和一个长度为 $N$ 的数列 $A$ 按如下方式决定。
首先,青木会得到一个有 $N$ 个面的骰子(每个面上分别标有 $1$ 到 $N$ 的整数,掷出每个面的概率相等),以及一个变量 $x=0$。
然后,重复以下步骤直到结束:
- 掷一次骰子,掷出的点数记为 $y$。
- 如果 $x < y$,则发放 $A_y$ 日元,并将 $x$ 更新为 $y$。
- 否则,流程结束。
青木本月的工资为通过上述流程累计发放的金额之和。
请计算青木本月工资的期望值,并对 $998244353$ 取模后输出。
期望值 $\bmod{998244353}$ 的定义
本题中要求的期望值一定是有理数。在本题的约束下,设期望值为最简分数 $\frac{y}{x}$,保证 $x$ 不会被 $998244353$ 整除。此时,存在唯一的 $0 \leq z < 998244353$ 满足 $y \equiv xz \pmod{998244353}$,请输出 $z$。
输入格式
输入通过标准输入给出,格式如下:
> $N$ $A_1$ $A_2$ $\dots$ $A_N$
输出格式
请输出答案。
说明/提示
### 数据范围
- 输入均为整数。
- $1 \leq N \leq 3 \times 10^5$
- $0 \leq A_i < 998244353$
### 样例解释 1
一种流程如下:
- 初始时,$x=0$。
- 掷一次骰子,结果为 $1$。由于 $0 < 1$,发放 $A_1=3$ 日元,$x$ 更新为 $1$。
- 再掷一次骰子,结果为 $3$。由于 $1 < 3$,发放 $A_3=6$ 日元,$x$ 更新为 $3$。
- 再掷一次骰子,结果为 $1$。由于 $3 \geq 1$,流程结束。
在这个例子中,青木本月的工资为 $9$ 日元。
本月工资的期望值可以计算为 $\frac{49}{9}$ 日元,模 $998244353$ 意义下为 $776412280$。
由 ChatGPT 4.1 翻译