結果
問題 |
No.1299 Random Array Score
|
ユーザー |
|
提出日時 | 2020-11-27 22:45:19 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 318 bytes |
コンパイル時間 | 364 ms |
コンパイル使用メモリ | 82,556 KB |
実行使用メモリ | 105,480 KB |
最終ジャッジ日時 | 2024-07-26 19:04:27 |
合計ジャッジ時間 | 15,027 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 WA * 1 |
ソースコード
import sys input = sys.stdin.readline N, K = map(int, input().split()) A = list(map(int, input().split())) if K==0: print(sum(A)) exit() MOD = 998244353 S = sum(A)*pow(N, K, MOD) for Ai in A: S = (S+N*(pow(2, K, MOD)-1)*Ai*pow(N, K-1, MOD))%MOD S = (S*pow(pow(N, K, MOD), MOD-2, MOD))%MOD print(S)