atcoder#ABC231G. [ABC231G] Balls in Boxes

[ABC231G] Balls in Boxes

题目描述

1 1 から N N の番号がついた N N 個の箱があります。最初、箱 i i には Ai A_i 個のボールが入っています。

あなたは次の操作を K K 回繰り返します。

  • N N 個の箱の中から等確率で 1 1 個選ぶ(この選択は操作ごとに独立である)。選んだ箱にボールを 1 1 個追加する。

K K 回の操作が終了した後で箱 i i に入っているボールの個数を Bi B_i とするとき、スコアはボールの個数の総積 i=1NBi \prod_{i=1}^{N}B_i になります。

スコアの期待値を mod 998244353 \bmod\ 998244353 で求めてください。

输入格式

入力は以下の形式で標準入力から与えられる。

N N K K A1 A_1 \ldots AN A_N

输出格式

答えを出力せよ。

题目大意

nn 个盒子,初始时第 ii 个盒子内有 aia_i 个小球,进行 kk 次操作后,每次操作等概率随机选择一个盒子放入一个小球,设 kk 次操作后每个盒子的小球个数为 bib_i,那么得分为 i=1nbi\prod_{i=1}^n b_i。求出期望得分。

  • n1000,k,ai109n\leq 1000,k,a_i\leq 10^9
3 1
1 2 3
665496245
2 2
1 2
499122182
10 1000000000
998244350 998244351 998244352 998244353 998244354 998244355 998244356 998244357 998244358 998244359
138512322

提示

注意

求める期待値が既約分数 p/q p/q で表されるとき、rq p (mod998244353) rq\equiv\ p\ \pmod{998244353} かつ 0 r < 998244353 0\leq\ r\ <\ 998244353 を満たす整数 r r がこの問題の制約のもとで一意に定まります。この r r が求める値です。

制約

  • 1  N  1000 1\ \leq\ N\ \leq\ 1000
  • 1  K  109 1\ \leq\ K\ \leq\ 10^9
  • 0  Ai  109 0\ \leq\ A_i\ \leq\ 10^9

Sample Explanation 1

操作の結果、スコアは次のようになります。 - 操作で箱 1 1 を選んだとき、2× 2× 3=12 2\times\ 2\times\ 3=12 - 操作で箱 2 2 を選んだとき、1× 3× 3=9 1\times\ 3\times\ 3=9 - 操作で箱 3 3 を選んだとき、1× 2× 4=8 1\times\ 2\times\ 4=8 したがって、求める期待値は 13(12+9+8)=293 \frac{1}{3}(12+9+8)=\frac{29}{3} となります。これを mod 998244353 \bmod\ 998244353 で表すと 665496245 665496245 になります。

Sample Explanation 2

操作の結果、スコアは次のようになります。 - 1 1 回目の操作で箱 1 1 を選び、2 2 回目の操作で箱 1 1 を選んだとき、3× 2=6 3\times\ 2=6 - 1 1 回目の操作で箱 1 1 を選び、2 2 回目の操作で箱 2 2 を選んだとき、2× 3=6 2\times\ 3=6 - 1 1 回目の操作で箱 2 2 を選び、2 2 回目の操作で箱 1 1 を選んだとき、2× 3=6 2\times\ 3=6 - 1 1 回目の操作で箱 2 2 を選び、2 2 回目の操作で箱 2 2 を選んだとき、1× 4=4 1\times\ 4=4 したがって、求める期待値は 14(6+6+6+4)=112 \frac{1}{4}(6+6+6+4)=\frac{11}{2} となります。