結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー shotoyoo
提出日時 2021-06-18 23:08:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,588 ms / 2,500 ms
コード長 702 bytes
コンパイル時間 297 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 76,288 KB
最終ジャッジ日時 2024-06-22 21:21:36
合計ジャッジ時間 18,673 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))


n,m = list(map(int, input().split()))
M = 998244353
ans = 0
inv2 = pow(2, M-2, M)
def f(v):
    return inv2 * (n * ((v+1) * pow(v, n, M) % M) % M) % M
def g(v):
    return inv2 * ((m + v) * (n * pow(m-v+1, n, M) % M) % M) % M
# for v in range(1,m+1):
#     ans += v * (f(v) - f(v-1)) % M
#     ans -= v * (g(v) - g(v+1)) % M
#     ans %= M
ans = m*f(m) - g(1)
for v in range(2, m+1):
    ans -= g(v)
    ans %= M
for v in range(1,m):
    ans -= f(v)
    ans %= M
print(ans%M)
0