AT_abc380_g [ABC380G] Another Shuffle Window
题目描述
[问题链接](https://atcoder.jp/contests/abc380/tasks/abc380_g)
给定一个排列 $P = (1, 2, \dots, N)$ 和一个整数 $K$。
请计算经过以下操作后,排列 $P$ 的**逆序对数**的期望值模 $998244353$ 的结果:
- 首先,从 $1$ 到 $N-K+1$ 的整数中随机均匀地选择一个整数 $i$;
- 然后,将子数组 $P_i, P_{i+1}, \dots, P_{i+K-1}$ 进行随机均匀打乱。
#### 逆序对是什么?
> 对于一个数组 $(A_1, A_2, \dots, A_N)$,逆序对是满足 $1 \leq i < j \leq N$ 且 $A_i > A_j$ 的整数对 $(i, j)$ 的个数。
#### 期望值模 $998244353$ 是什么?
> 在本问题的约束条件下,期望值可以表示为一个分数 $\frac{P}{Q}$,且 $Q \not \equiv 0 \pmod{998244353}$。
> 因此可以找到一个唯一的整数 $R$ 满足:
> $$
> R \times Q \equiv P \pmod{998244353}, \quad 0 \leq R < 998244353
> $$
> 你需要输出这个整数 $R$。
输入格式
从标准输入流读入,格式如下:
$$
N\ K\ P_1\ P_2\ \dots\ P_N
$$
输出格式
输出在标准输出流,在一行中输出答案。
说明/提示
#### 约束条件
- 所有输入均为整数;
- $1 \leq K \leq N \leq 2 \times 10^5$;
- $P$ 是 $(1, 2, \dots, N)$ 的一个排列。
#### 样例解释 1
通过操作,排列 $P$ 会变为以下形式:
- $(1, 4, 2, 3)$ —— 概率 $\frac{1}{2}$;
- $(4, 1, 2, 3)$ —— 概率 $\frac{1}{6}$;
- $(1, 2, 4, 3)$ —— 概率 $\frac{1}{6}$;
- $(1, 4, 3, 2)$ —— 概率 $\frac{1}{6}$。
对应的逆序对数期望值为:
$$
\displaystyle 2 \times \frac{1}{2} + 3 \times \frac{1}{6} + 1 \times \frac{1}{6} + 3 \times \frac{1}{6} = \frac{13}{6}
$$
将 $\frac{13}{6}$ 转换为模 $998244353$ 的结果为 $166374061$,因此输出 $166374061$。
Translated By [$\mathtt{Mr\_Az}$](/user/536560)。