AT_abc295_e [ABC295E] Kth Number
题目描述
有一个长度为 $N$ 的数列 $A=(A_1,A_2,\dots,A_N)$,其中每个元素都是 $0$ 到 $M$ 之间的整数。
现在,すぬけくん将依次进行以下两个操作:
1. 对于所有满足 $A_i=0$ 的 $i$,独立且等概率地选择一个 $1$ 到 $M$ 之间的整数,将 $A_i$ 替换为该整数。
2. 将数列 $A$ 按升序排序。
请输出すぬけくん完成操作 1 和 2 后 $A_K$ 的期望值,结果对 $998244353$ 取模。
“对 $998244353$ 取模输出期望值” 的意思是:可以证明,所求期望值一定是有理数。在本题的约束下,设其为 $\frac{P}{Q}$,其中 $P$ 和 $Q$ 互质。则存在唯一的整数 $R$ 满足 $R \times Q \equiv P \pmod{998244353}$ 且 $0 \leq R < 998244353$。请输出这个 $R$。
输入格式
输入按以下格式从标准输入读入。
> $N$ $M$ $K$ $A_1$ $A_2$ $\dots$ $A_N$
输出格式
请输出すぬけくん完成操作 1 和 2 后 $A_K$ 的期望值,对 $998244353$ 取模。
说明/提示
## 约束
- $1\leq K\leq N\leq 2000$
- $1\leq M\leq 2000$
- $0\leq A_i\leq M$
- 输入均为整数
## 样例解释 1
すぬけくん在操作 1 中将 $A_2$ 替换为 $1$ 到 $5$ 之间的整数。设该整数为 $x$,则:
- 当 $x=1,2$ 时,操作 1 和 2 后 $A_2=2$。
- 当 $x=3$ 时,操作 1 和 2 后 $A_2=3$。
- 当 $x=4,5$ 时,操作 1 和 2 后 $A_2=4$。
因此,$A_2$ 的期望值为 $\frac{2+2+3+4+4}{5}=3$。
## 样例解释 2
期望值为 $\frac{14}{9}$。
由 ChatGPT 4.1 翻译