結果

問題 No.2113 Distance Sequence 1.5
ユーザー ああいい
提出日時 2022-10-31 12:44:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 374 bytes
コンパイル時間 145 ms
コンパイル使用メモリ 81,844 KB
実行使用メモリ 53,488 KB
最終ジャッジ日時 2024-07-08 02:11:27
合計ジャッジ時間 1,909 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M,K = map(int,input().split())
N *= 2
P = 998244353

import sys
if M < K:
    print(pow(M,N,P))
    exit()

ans = 0
if K == 1:
    print(M%P)
    exit()
if K == 2:
    ans = (M - 1) * pow(2,N - 2,P) + M
    print(ans%P)
    exit()

inv = pow(K-2,P-2,P)
tmp = pow(K - 1,N,P) - 1
tmp = pow(K,N,P)-pow(K-1,N,P)
ans = tmp * (M - K) % P

t = pow(K,N,P)

ans += t
print(ans%P)
0