AT_awtf2024_b 01 Inversion Expected
题目描述
给定一个由 `0` 和 `1` 组成的长度为 $N$ 的字符串 $S$。对于满足 $1 \leq i < j \leq N$,且 $S$ 的第 $i$ 个字符为 `1`,第 $j$ 个字符为 `0` 的整数对 $(i, j)$,我们称其为**逆序对**。
只要 $S$ 中存在逆序对,就进行如下操作:
- 随机选择一个逆序对 $(i, j)$。每次选择都是独立且等概率的。然后交换 $S$ 的第 $i$ 个字符和第 $j$ 个字符。
请计算操作次数的期望值,并对 $998244353$ 取模后输出。
期望值 $\bmod\ 998244353$ 的定义:可以证明所求期望值一定是有理数。在本题的约束下,将其表示为最简分数 $\frac{P}{Q}$ 时,$Q \not\equiv 0 \pmod{998244353}$ 也可以保证。因此,存在唯一的整数 $R$ 满足 $R \times Q \equiv P \pmod{998244353},\ 0 \leq R < 998244353$。请输出 $R$。
输入格式
输入以以下格式从标准输入读入。
> $N$ $S$
输出格式
请输出答案。
说明/提示
## 限制
- $1 \leq N \leq 250000$
- $S$ 是由 `0` 和 `1` 组成的长度为 $N$ 的字符串
## 样例解释 1
操作次数的期望值为 $1$。
## 样例解释 2
操作次数的期望值为 $3/2$。
由 ChatGPT 4.1 翻译