結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー H3PO4
提出日時 2022-10-15 09:50:54
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 488 bytes
コンパイル時間 222 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 86,400 KB
最終ジャッジ日時 2024-06-26 19:36:16
合計ジャッジ時間 4,518 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 TLE * 1
other -- * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, M = map(int, input().split())
MOD = 998244353


def tri(x):
    return x * (x + 1) // 2


def fix_max(k):
    return k * N * (tri(k) * pow(k, N - 1, MOD) - tri(k - 1) * pow(k - 1, N - 1, MOD)) % MOD


def fix_min(k):
    return k * N * ((tri(M) - tri(k - 1)) * pow(M - k + 1, N - 1, MOD) - (tri(M) - tri(k)) * pow(M - k, N - 1, MOD)) % MOD


ans = 0
for k in range(1, M + 1):
    # print(k, fix_max(k), fix_min(k))
    ans += fix_max(k)
    ans -= fix_min(k)
    ans %= MOD
print(ans)
0