AT_abc295_e.[ABC295E] Kth Number
省选/NOI-
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
有一个长度为 N 的数列 A=(A1,A2,…,AN),其中每个元素都是 0 到 M 之间的整数。
现在,すぬけくん将依次进行以下两个操作:
- 对于所有满足 Ai=0 的 i,独立且等概率地选择一个 1 到 M 之间的整数,将 Ai 替换为该整数。
- 将数列 A 按升序排序。
请输出すぬけくん完成操作 1 和 2 后 AK 的期望值,结果对 998244353 取模。
“对 998244353 取模输出期望值” 的意思是:可以证明,所求期望值一定是有理数。在本题的约束下,设其为 QP,其中 P 和 Q 互质。则存在唯一的整数 R 满足 R×Q≡P(mod998244353) 且 0≤R<998244353。请输出这个 R。
输入格式
输入按以下格式从标准输入读入。
N M K A1 A2 … AN
输出格式
请输出すぬけくん完成操作 1 和 2 后 AK 的期望值,对 998244353 取模。
输入输出样例
输入#1
3 5 2 2 0 4
输出#1
3
输入#2
2 3 1 0 0
输出#2
221832080
输入#3
10 20 7 6 5 0 2 0 0 0 15 0 0
输出#3
617586310
说明/提示
约束
- 1≤K≤N≤2000
- 1≤M≤2000
- 0≤Ai≤M
- 输入均为整数
样例解释 1
すぬけくん在操作 1 中将 A2 替换为 1 到 5 之间的整数。设该整数为 x,则:
- 当 x=1,2 时,操作 1 和 2 后 A2=2。
- 当 x=3 时,操作 1 和 2 后 A2=3。
- 当 x=4,5 时,操作 1 和 2 后 A2=4。
因此,A2 的期望值为 52+2+3+4+4=3。
样例解释 2
期望值为 914。
由 ChatGPT 4.1 翻译
输入解题思路,AI测评打分。不知道怎么写?