AT_abc249_h.[ABC249Ex] Dye Color

NOI/NOI+/CTSC

通过率:0%

AC君温馨提醒

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

题目描述

有 NN 个球,每个球编号为 11 到 NN。最初,第 ii 个球被涂上颜色 AiA_i。

颜色用 11 到 NN 之间的整数表示。

你需要重复以下操作,直到所有球的颜色都相同为止:

  • NN 个球的所有子集(包括空集)共有 2N2^N 个,从中等概率随机选择一个集合。设选中的集合包含的球的编号按升序为 X1,X2,…,XKX_1, X_2, \dots, X_K。然后,从 1,2,…,N1,2,\dots,N 中选取 KK 个不同的数,得到一个排列,等概率随机选择一个排列 P=(P1,P2,…,PK)P=(P_1,P_2,\dots,P_K)。接下来,对于每个 1≤i≤K1\le i\le K,将球 XiX_i 的颜色改为 PiP_i。

请计算使所有球颜色相同所需操作次数的期望值,结果对 998244353998244353 取模。

这里,从 (1,2,…,N)(1,2,\dots,N) 中选取 KK 个不同的数得到的排列,指的是由 KK 个 11 到 NN 之间互不相同的整数构成的序列。

输入格式

输入为一行,格式如下:

N A1 A2 … ANN\ A_1\ A_2\ \dots\ A_N

输出格式

输出答案。

输入输出样例

  • 输入#1

    2
    1 2

    输出#1

    4
  • 输入#2

    3
    1 1 1

    输出#2

    0
  • 输入#3

    10
    3 1 4 1 5 9 2 6 5 3

    输出#3

    900221128

说明/提示

注释

可以证明,所求的期望值一定是有理数。在本题的约束下,若用互质的两个整数 P,QP, Q 表示为 PQ\frac{P}{Q},则存在唯一的整数 RR 满足 R×Q≡P(mod998244353)R\times Q\equiv P\pmod{998244353} 且 0≤R<9982443530\le R<998244353。请输出这个 RR。

数据范围

  • 2≤N≤20002\le N\le 2000
  • 1≤Ai≤N1\le A_i\le N
  • 输入均为整数。

样例解释 1

每次选择大小为 11 的集合,并且操作会一直持续,直到将未选球的颜色改为与之相同。该概率为 24×12=14\frac{2}{4}\times\frac{1}{2}=\frac{1}{4},因此期望操作次数为 44。

样例解释 2

由于一开始所有球的颜色都相同,因此不需要进行任何操作。

由 ChatGPT 4.1 翻译

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

首页