結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
|
提出日時 | 2021-06-18 22:06:59 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 449 bytes |
コンパイル時間 | 359 ms |
コンパイル使用メモリ | 82,104 KB |
実行使用メモリ | 83,152 KB |
最終ジャッジ日時 | 2024-06-22 20:29:13 |
合計ジャッジ時間 | 4,335 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 TLE * 1 |
other | -- * 20 |
ソースコード
MOD = 998244353 N, M = map(int, input().split()) ans = 0 def s(x): return (x * (x + 1) // 2) % MOD for m in range(1, M + 1): tmp = s(m) * pow(m, N - 1, MOD) - s(m - 1) * pow(m - 1, N - 1, MOD) tmp %= MOD ans = (ans + m * tmp % MOD * N) % MOD for m in range(1, M + 1): tmp = (s(M) - s(m - 1)) * pow(M - m + 1, N - 1, MOD) - (s(M) - s(m)) * pow(M - m, N - 1, MOD) tmp %= MOD ans = (ans - m * tmp % MOD * N) % MOD print(ans)