結果
問題 | No.1552 Simple Dice Game |
ユーザー | Kude |
提出日時 | 2021-06-18 23:16:58 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 668 bytes |
コンパイル時間 | 431 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 92,376 KB |
最終ジャッジ日時 | 2024-06-22 21:26:12 |
合計ジャッジ時間 | 29,992 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 TLE * 2 |
ソースコード
MOD = 998244353 fact_range = 10**6 fact = [1] * (fact_range + 1) for i in range(0, fact_range): fact[i+1] = fact[i] * (i + 1) % MOD ifact = [1] * (fact_range + 1) ifact[fact_range] = pow(fact[fact_range], MOD - 2, MOD) for i in range(fact_range, 0, -1): ifact[i-1] = ifact[i] * i % MOD def comb(n, k): if k < 0 or n < k: return 0 else: return fact[n] * ifact[n-k] % MOD * ifact[k] % MOD n, m = map(int, input().split()) ans = 0 x = (1 + m) * pow(2, MOD - 2, MOD) % MOD for d in range(m): ans += d * (m - d) * (pow(d + 1, n, MOD) - 2 * pow(d, n, MOD) + (pow(d - 1, n, MOD) if d - 1 >= 0 else 0)) * x * n ans %= MOD print(ans)