結果
問題 |
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)