atcoder#ARC133F. [ARC133F] Random Transition

[ARC133F] Random Transition

配点 : 11001100

問題文

整数 NN が与えられます.

すぬけくんはこれから,以下の操作を行います.

  • 変数 xx を用意し,ランダムに選んだ 00 以上 NN 以下の整数で初期化する.各 0iN0 \leq i \leq N に対し,x=ix=i と初期化する確率は Ai/109A_i/10^9 である.
  • 次の操作を KK 回繰り返す.- 確率 x/Nx/Nxx の値を 11 減らし,確率 1x/N1-x/Nxx の値を 11 増やす. (ここで,操作後も xx の値が必ず 00 以上 NN 以下であることに注意)
  • 確率 x/Nx/Nxx の値を 11 減らし,確率 1x/N1-x/Nxx の値を 11 増やす. (ここで,操作後も xx の値が必ず 00 以上 NN 以下であることに注意)

i=0,1,,Ni=0,1,\cdots,N について,すべての操作が終了したあとに x=ix=i である確率を (mod998244353)\pmod{998244353} で求めてください.

確率 $\pmod{998244353}$ の定義

求める確率は必ず有理数になることが証明できます。 また、この問題の制約のもとでは、その値を既約分数 PQ\frac{P}{Q} で表した時、Q0(mod998244353)Q \neq 0 \pmod{998244353} となることも証明できます。 よって、$R \times Q \equiv P \pmod{998244353}, 0 \leq R < 998244353$ を満たす整数 RR が一意に定まります。 この RR を答えてください。

制約

  • 1N1000001 \leq N \leq 100000
  • 0Ai0 \leq A_i
  • 0iNAi=109\sum_{0 \leq i \leq N}A_i = 10^9
  • 1K1091 \leq K \leq 10^9
  • 入力される値はすべて整数である

入力

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

NN KK

A0A_0 A1A_1 \cdots ANA_N

出力

i=0,1,,Ni=0,1,\cdots,N について,すべての操作が終了したあとに x=ix=i である確率を (mod998244353)\pmod{998244353} で出力せよ.

2 1
0 1000000000 0
499122177 0 499122177

最初は必ず x=1x=1 と初期化します. その後の操作では.確率 1/21/2xx の値を 11 減らし,確率 1/21/2xx の値を 11 増やします. 最終的に x=0,1,2x=0,1,2 となる確率は,それぞれ 1/2,0,1/21/2,0,1/2 です.

4 2
200000000 200000000 200000000 200000000 200000000
723727156 598946612 349385524 598946612 723727156
10 100
21265166 263511538 35931763 26849698 108140810 134702248 36774526 147099145 58335759 4118743 163270604
505314898 24510700 872096939 107940764 808162829 831195162 314651262 535843032 665830283 627881537 696038713