AT_acl1_e Shuffle Window
题目描述
给定一个 $ (1, 2, \dots, N) $ 的排列 $ p_1, p_2, \dots, p_N $ 和一个整数 $ K $。maroon 君会对 $ i = 1, 2, \dots, N - K + 1 $ 依次进行如下操作:
- 将 $ p_i, p_{i+1}, \dots, p_{i+K-1} $ 这 $ K $ 个数进行等概率随机打乱。
请你求出所有操作结束后,数列的逆序对数的期望值,并将结果对 $ 998244353 $ 取模后输出。
更准确地说,期望值可以表示为一个最简分数 $ \frac{P}{Q} $,并且一定存在唯一的整数 $ R $ 满足 $ R \times Q \equiv P \pmod{998244353},\ 0 \leq R < 998244353 $。请输出这个 $ R $。
另外,数列 $ a_1, a_2, \dots, a_N $ 的逆序对数定义为满足 $ i < j,\ a_i > a_j $ 的有序对 $ (i, j) $ 的个数。
输入格式
> $ N $ $ K $ $ p_1 $ $ p_2 $ ... $ p_N $
输出格式
请输出期望值对 $ 998244353 $ 取模后的结果。
说明/提示
### 限制条件
- $ 2 \leq N \leq 200,\!000 $
- $ 2 \leq K \leq N $
- $ (p_1, p_2, \dots, p_N) $ 是 $ (1, 2, \dots, N) $ 的一个排列
- 输入的所有数均为整数。
### 样例解释 1
最终的数列可能为 $ (1, 2, 3) $、$ (2, 1, 3) $、$ (1, 3, 2) $、$ (2, 3, 1) $,它们出现的概率均为 $ \frac{1}{4} $。这些数列的逆序对数分别为 $ 0, 1, 1, 2 $,因此期望值为 $ 1 $。
由 ChatGPT 4.1 翻译