結果

問題 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,185 ms / 2,500 ms
+ 541µs
コード長 546 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 253 ms
コンパイル使用メモリ 95,980 KB
実行使用メモリ 120,320 KB
最終ジャッジ日時 2026-07-27 09:28:35
合計ジャッジ時間 16,034 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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