結果

問題 No.1552 Simple Dice Game
コンテスト
ユーザー H3PO4
提出日時 2022-10-15 09:50:54
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 1,208 ms / 2,500 ms
+ 561µs
コード長 488 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 92 ms
コンパイル使用メモリ 64,948 KB
実行使用メモリ 67,808 KB
最終ジャッジ日時 2026-10-20 07:22:23
合計ジャッジ時間 15,644 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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