AT_abc450_g [ABC450G] Random Subtraction

题目描述

给定一个长度为 $N$ 的非负整数序列 $A$。 重复以下操作,直到 $A$ 只剩下一个元素为止: - 等概率随机选择两个不相同的整数 $i$ 和 $j$,满足 $1 \leq i, j \leq |A|$。 - 令 $a = A_i$,$b = A_j$。 - 从 $A$ 中移除第 $i$ 和第 $j$ 个元素。 - 将 $a - b$ 添加到 $A$ 的末尾。 设最终 $A$ 中唯一的元素为 $x$。求 $x^2$ 的期望值对 $998244353$ 取模后的结果。 期望值对 $998244353$ 取模的定义:可以证明需要计算的期望值总是一个有理数。也已知,在本题的约束下,若期望值记为最简分数 $\frac{P}{Q}$,则有 $Q \not\equiv 0 \pmod{998244353}$。因此,存在唯一的整数 $R$,满足 $R \times Q \equiv P \pmod{998244353}$,且 $0 \leq R < 998244353$。请输出这个 $R$。

输入格式

输入采用标准输入方式,格式如下: > $N$ $A_1$ $A_2$ $\dots$ $A_N$

输出格式

输出一行,即答案。

说明/提示

### 样例解释 1 首先,选择 $(i, j) = (3, 1)$,序列变为 $(5, -4)$。 接着,选择 $(i, j) = (1, 2)$,序列变为 $(9)$。 此时 $x^2$ 以概率 $\frac{1}{3}$ 取 $81$,以概率 $\frac{2}{3}$ 取 $1$,所以期望值为 $\frac{83}{3}$。 ### 约束条件 - $1 \leq N \leq 2 \times 10^5$ - $0 \leq A_i \leq 998244352$ - 所有输入值均为整数。 由 ChatGPT 5 翻译