AT_abc295_e.[ABC295E] Kth Number

省选/NOI-

通过率:0%

AC君温馨提醒

该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

有一个长度为 NN 的数列 A=(A1,A2,…,AN)A=(A_1,A_2,\dots,A_N),其中每个元素都是 00 到 MM 之间的整数。

现在,すぬけくん将依次进行以下两个操作:

  1. 对于所有满足 Ai=0A_i=0 的 ii,独立且等概率地选择一个 11 到 MM 之间的整数,将 AiA_i 替换为该整数。
  2. 将数列 AA 按升序排序。

请输出すぬけくん完成操作 1 和 2 后 AKA_K 的期望值,结果对 998244353998244353 取模。

“对 998244353998244353 取模输出期望值” 的意思是:可以证明,所求期望值一定是有理数。在本题的约束下,设其为 PQ\frac{P}{Q},其中 PP 和 QQ 互质。则存在唯一的整数 RR 满足 R×Q≡P(mod998244353)R \times Q \equiv P \pmod{998244353} 且 0≤R<9982443530 \leq R < 998244353。请输出这个 RR。

输入格式

输入按以下格式从标准输入读入。

NN MM KK A1A_1 A2A_2 …\dots ANA_N

输出格式

请输出すぬけくん完成操作 1 和 2 后 AKA_K 的期望值,对 998244353998244353 取模。

输入输出样例

  • 输入#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≤20001\leq K\leq N\leq 2000
  • 1≤M≤20001\leq M\leq 2000
  • 0≤Ai≤M0\leq A_i\leq M
  • 输入均为整数

样例解释 1

すぬけくん在操作 1 中将 A2A_2 替换为 11 到 55 之间的整数。设该整数为 xx,则:

  • 当 x=1,2x=1,2 时,操作 1 和 2 后 A2=2A_2=2。
  • 当 x=3x=3 时,操作 1 和 2 后 A2=3A_2=3。
  • 当 x=4,5x=4,5 时,操作 1 和 2 后 A2=4A_2=4。

因此,A2A_2 的期望值为 2+2+3+4+45=3\frac{2+2+3+4+4}{5}=3。

样例解释 2

期望值为 149\frac{14}{9}。

由 ChatGPT 4.1 翻译

输入解题思路,AI测评打分。不知道怎么写?

首页