問題一覧 > 通常問題

No.2793 Mancala Master

レベル : / 実行時間制限 : 1ケース 2.000秒 / メモリ制限 : 512 MB / 標準ジャッジ問題
タグ : / 解いたユーザー数 19
作問者 : 箱星 / テスター : dadas
2 ProblemId : 10465 / 出題時の順位表 / 自分の提出
問題文最終更新日: 2024-06-23 08:40:59

問題文

正整数列 A=(A1,,AN)A=(A_1,\ldots,A_N) と正整数 KK が与えられます。

小星さんは次のようなゲームを行います。はじめスコアは 11 とし、i=1,2,,Ni=1,2,\ldots,N に対して座標 ii に石を AiA_i 個置きます。

i=N,N1,,1i=N,N-1,\ldots,1 の順に次の操作を行います。

  • 座標 ii にあるすべての石を座標 0,1,,i10,1,\ldots,i-1 のいずれかに動かす。座標 jj に動かす石の個数を cjc_j とするとき、スコアに f(c0)f(c1)f(ci1)f(c_0)f(c_1)\cdots f(c_{i-1}) をかける。ここで f(m)={KmKm1(m>0)1(m=0) f(m)=\begin{cases} K^m-K^{m-1} & (m>0) \\ 1 & (m=0) \end{cases} である。

可能な操作方法すべてについての最終的なスコアの総和を 998244353998244353 で割った余りを求めてください。

TT 個のテストケースが与えられるので、それぞれについて答えてください。

制約

  • 1T1051\le T\le 10^5
  • 1N2×1051\le N\le 2\times 10^5
  • 1Ai<9982443531\le A_i\lt 998244353
  • 2K<9982443532\le K\lt 998244353
  • 入力はすべて整数
  • 11 つの入力に含まれるテストケースについて、NN の総和は 2×1052\times 10^5 以下

入力

入力は以下の形式で標準入力から与えられます。
TT
case1\mathrm{case}_1
\vdots
caseT\mathrm{case}_T

各ケースは以下の形式で与えられます。

NN KK
A1A_1 A2A_2 \cdots ANA_N

出力

TT 行出力してください。ii 行目には ii 番目のテストケースに対する答えを出力してください。

サンプル

サンプル1
入力
1
2 10
1 2
出力
89100

まず座標 22 にある 22 個の石を動かします。

  • 22 個とも座標 00 に動かしたとき、スコアに 10210=9010^2-10=90 をかけます。次に座標 11 にある 11 個の石を座標 00 に動かし、スコアに 101=910-1=9 をかけます。
  • 22 個とも座標 11 に動かしたとき、スコアに 10210=9010^2-10=90 をかけます。次に座標 11 にある 33 個の石を座標 00 に動かし、スコアに 103102=90010^3-10^2=900 をかけます。
  • 11 個ずつ座標 0,10,1 に動かしたとき、スコアに (101)(101)=81(10-1)(10-1)=81 をかけます。次に座標 11 にある 22 個の石を座標 00 に動かし、スコアに 10210=9010^2-10=90 をかけます。

よってスコアの総和は 90×9+90×900+81×90=8910090\times 9+90\times 900+81\times 90=89100 です。

サンプル2
入力
2
1 2
3
3 3281572
842184 12484 58238890
出力
4
724046312

998244353998244353 で割った余りを求めてください。

提出するには、Twitter 、GitHub、 Googleもしくは右上の雲マークをクリックしてアカウントを作成してください。