結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー 👑 SPD_9X2
提出日時 2021-06-18 22:36:39
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 546 bytes
コンパイル時間 394 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 108,800 KB
最終ジャッジ日時 2024-06-22 21:01:32
合計ジャッジ時間 34,112 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14 TLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

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