結果

問題 No.1552 Simple Dice Game
ユーザー KudeKude
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0