[ABC249Ex] Dye Color

题意翻译

有 $n$ 个球,编号 $1\sim n$,初始时球 $i$ 的颜色为 $A_i$。颜色为 $1\sim n$ 的整数。 进行以下操作若干次,直到所有球颜色相同: * 均匀随机地选择一个球 $1\sim n$ 的子集(可以为空),记该集合中的球的编号为 $X_1,X_2,\cdots,X_k$。 * 均匀随机地选择一个长为 $k$ 的数列,其中每个元素为 $1\sim n$ 中的一个数,且两两不同,记这个数列为 $P_1,P_2,\cdots,P_k$。 * $\forall 1\le i\le k$,将球 $X_i$ 染成颜色 $P_i$。 求操作次数的期望对 $998244353$ 取模后的值。

题目描述

[problemUrl]: https://atcoder.jp/contests/abc249/tasks/abc249_h $ N $ 個のボールがあり、ボールには $ 1 $ から $ N $ の番号がついています。初め、ボール $ i $ は色 $ A_i $ で塗られています。 色は $ 1 $ 以上 $ N $ 以下の整数によって表されます。 以下の操作を、全てのボールの色が等しくなるまで繰り返します。 - $ N $ 個のボールからなる集合の部分集合(空集合を含む)は $ 2^N $ 個あるが、そこから $ 1 $ 個の集合を一様ランダムに選ぶ。選んだ集合に含まれるボールの index を昇順に $ X_1,X_2,...,X_K $ とする。次に、$ (1,2,\dots,N) $ から $ K $ 個を選んで得られる順列を一様ランダムに $ 1 $ 個選び $ P=(P_1,P_2,\dots,P_K) $ とする。そして、$ 1\ \le\ i\ \le\ K $ を満たす整数 $ i $ に対し、ボール $ X_i $ の色を $ P_i $ に変更する。 操作回数の期待値 $ \bmod\ 998244353 $ を求めてください。 ここで、$ (1,2,\dots,N) $ から $ K $ 個を選んで得られる順列とは、$ 1 $ 以上 $ N $ 以下の整数 $ K $ 個からなる数列であって、要素が相異なるもののことです。

输入输出格式

输入格式


入力は以下の形式で標準入力から与えられる。 > $ N $ $ 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

说明

### 注記 求める期待値は必ず有理数となることが証明できます。またこの問題の制約下では、その値を互いに素な $ 2 $ つの整数 $ P,Q $ を用いて $ \frac{P}{Q} $ と表した時、$ R\ \times\ Q\ \equiv\ P(\bmod\ 998244353) $ かつ $ 0\ \le\ R\ <\ 998244353 $ を満たす整数 $ R $ がただ一つ存在することが証明できます。この $ R $ を求めてください。 ### 制約 - $ 2\ \le\ N\ \le\ 2000 $ - $ 1\ \le\ A_i\ \le\ N $ - 入力は全て整数である。 ### Sample Explanation 1 大きさが $ 1 $ である集合を選び、かつ集合に含まれないもう一方のボールの色に変更するまで操作は続きます。その確率は $ \displaystyle\ \frac{2}{4}\ \times\ \frac{1}{2}=\frac{1}{4} $ なので、操作回数の期待値は $ 4 $ 回です。 ### Sample Explanation 2 初めから全てのボールの色が同じであるため、操作は行われません。