結果
問題 | No.1552 Simple Dice Game |
ユーザー |
|
提出日時 | 2021-06-18 22:04:54 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 399 bytes |
コンパイル時間 | 235 ms |
コンパイル使用メモリ | 82,292 KB |
実行使用メモリ | 78,160 KB |
最終ジャッジ日時 | 2024-06-22 20:26:14 |
合計ジャッジ時間 | 4,392 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 TLE * 1 |
other | AC * 13 TLE * 7 |
ソースコード
MOD = 998244353 N, M = map(int, input().split()) ans = 0 def s(x): return x * (x + 1) // 2 for m in range(1, M + 1): tmp = s(m) * pow(m, N - 1, MOD) - s(m - 1) * pow(m - 1, N - 1, MOD) ans = (ans + m * tmp * 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) ans = (ans - m * tmp * N) % MOD print(ans)