結果
問題 |
No.1552 Simple Dice Game
|
ユーザー |
![]() |
提出日時 | 2021-06-18 22:39:08 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,106 ms / 2,500 ms |
コード長 | 752 bytes |
コンパイル時間 | 244 ms |
コンパイル使用メモリ | 82,536 KB |
実行使用メモリ | 79,976 KB |
最終ジャッジ日時 | 2024-06-22 21:03:28 |
合計ジャッジ時間 | 17,634 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 20 |
ソースコード
n, m = map(int, input().split()) MOD = 998244353 inv2 = pow(2, MOD - 2, MOD) pows = [1] * (5 * 10 ** 5 + 10) for i in range(5 * 10 ** 5 + 10): pows[i] = pow(i, n, MOD) def sum_max(k): ans = k # 最大 r = pows[k] * (k + 1) * inv2 * n l = pows[k - 1] * k * inv2 * n ans *= r - l return ans % MOD def sum_min(k): ans = k # 最小 r = pows[m - k + 1] * (k + m) * inv2 * n l = pows[m - k] * (k + m + 1) * inv2 * n ans *= r - l return ans % MOD ans = 0 for k in range(1, m + 1): ans += sum_max(k) ans -= sum_min(k) ans %= MOD print(ans) """ import itertools tmp = 0 for ptn in itertools.product([i + 1 for i in range(m)], repeat=n): tmp += sum(ptn) * max(ptn) tmp %= MOD print(tmp)"""