結果
| 問題 |
No.1552 Simple Dice Game
|
| コンテスト | |
| ユーザー |
rlangevin
|
| 提出日時 | 2023-08-08 12:45:05 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 1,202 ms / 2,500 ms |
| コード長 | 393 bytes |
| コンパイル時間 | 496 ms |
| コンパイル使用メモリ | 82,176 KB |
| 実行使用メモリ | 80,256 KB |
| 最終ジャッジ日時 | 2024-11-08 18:25:40 |
| 合計ジャッジ時間 | 15,610 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 20 |
ソースコード
N, M = map(int, input().split())
mod = 998244353
ans = 0
coef = [0] * (M + 1)
for i in range(1, M + 1):
coef[i] = pow(i, N - 1, mod)
for i in range(1, M + 1):
ans += i * (i*(i+1)//2 * coef[i] * N)
ans -= i * (i*(i-1)//2 * coef[i-1] * N)
ans += i * ((M*(M+1)//2-i*(i+1)//2) * coef[M-i] * N)
ans -= i * ((M*(M+1)//2-i*(i-1)//2) * coef[M-i+1] * N)
ans %= mod
print(ans)
rlangevin