結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
|
提出日時 | 2022-10-15 09:50:54 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 488 bytes |
コンパイル時間 | 222 ms |
コンパイル使用メモリ | 82,688 KB |
実行使用メモリ | 86,400 KB |
最終ジャッジ日時 | 2024-06-26 19:36:16 |
合計ジャッジ時間 | 4,518 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 TLE * 1 |
other | -- * 20 |
ソースコード
N, M = map(int, input().split()) MOD = 998244353 def tri(x): return x * (x + 1) // 2 def fix_max(k): return k * N * (tri(k) * pow(k, N - 1, MOD) - tri(k - 1) * pow(k - 1, N - 1, MOD)) % MOD def fix_min(k): return k * N * ((tri(M) - tri(k - 1)) * pow(M - k + 1, N - 1, MOD) - (tri(M) - tri(k)) * pow(M - k, N - 1, MOD)) % MOD ans = 0 for k in range(1, M + 1): # print(k, fix_max(k), fix_min(k)) ans += fix_max(k) ans -= fix_min(k) ans %= MOD print(ans)