結果
問題 | 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 = 998244353inv2 = 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 * nl = pows[k - 1] * k * inv2 * nans *= r - lreturn ans % MODdef sum_min(k):ans = k # 最小r = pows[m - k + 1] * (k + m) * inv2 * nl = pows[m - k] * (k + m + 1) * inv2 * nans *= r - lreturn ans % MODans = 0for k in range(1, m + 1):ans += sum_max(k)ans -= sum_min(k)ans %= MODprint(ans)"""import itertoolstmp = 0for ptn in itertools.product([i + 1 for i in range(m)], repeat=n):tmp += sum(ptn) * max(ptn)tmp %= MODprint(tmp)"""