結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー 👑 SPD_9X2
提出日時 2021-06-18 22:36:39
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 1,147 ms / 2,500 ms
コード長 546 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 173 ms
コンパイル使用メモリ 85,348 KB
実行使用メモリ 109,340 KB
最終ジャッジ日時 2026-03-07 03:25:19
合計ジャッジ時間 14,045 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from sys import stdin
import sys
import math

mod = 998244353
def inverse(x):
    return pow(x,mod-2,mod)
half = inverse(2)


N,M = map(int,stdin.readline().split())
ans = 0

dp = []

nowok = 0

for K in range(M):

    a = M-K
    if K != 0:
        nowok += pow(K,N,mod) - pow(K-1,N,mod)
    else:
        nowok += pow(K,N,mod)
    nowok %= mod

    other = (pow(K+1,N,mod) - pow(K,N,mod)) * (M-K)

    dp.append((nowok + other) % mod)

dp.append(0)
ans = 0
for K in range(M):
    ans += (dp[K]-dp[K-1]) * K

print (ans * (M+1) * N * half % mod)
0