結果
| 問題 |
No.2113 Distance Sequence 1.5
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2022-10-28 23:57:07 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 38 ms / 2,000 ms |
| コード長 | 455 bytes |
| コンパイル時間 | 580 ms |
| コンパイル使用メモリ | 82,560 KB |
| 実行使用メモリ | 52,608 KB |
| 最終ジャッジ日時 | 2024-07-06 03:15:13 |
| 合計ジャッジ時間 | 2,083 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 21 |
ソースコード
N,M,K = map(int,input().split())
if K == 1:
print(M)
exit()
mod = 998244353
def pow_k(x, n,mod):
"""
O(log n)
"""
if n == 0:
return 1
K = 1
while n > 1:
if n % 2 != 0:
K = (K*x) % mod
x = (x*x) % mod
n //= 2
return (K * x)%mod
if M <= K:
ans = pow_k(M,2*N,mod)
print(ans)
else:
ans = pow_k(K,2*N,mod)
ss = pow_k(K-1,2*N,mod)
print((ans+(M-K)*(ans-ss))%mod)