結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
  
  
0