結果

問題 No.2156 ぞい文字列
ユーザー tomoyawatomoyawa
提出日時 2023-01-02 20:22:59
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 481 ms / 2,000 ms
コード長 424 bytes
コンパイル時間 421 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 44,216 KB
最終ジャッジ日時 2024-11-27 01:15:10
合計ジャッジ時間 12,513 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
import math
M = 998244353
def fibmtx2n(k):   # n = 2 ** n2
  fm = np.array([[1,1],[1,0]], dtype='object')
  for i in range(k): 
    fm = np.dot(fm,fm) % M
  return fm
def fibmtx(n):
  fm1 = np.array([[1,0],[0,1]],dtype='object')
  while n > 0:
    l2n = n.bit_length()-1
    fm1 = np.dot(fm1, fibmtx2n(l2n)) % M
    n -= 2**l2n
  return fm1[0][1]

N=int(input())
ans=(fibmtx(N)+fibmtx(N-1)-1)%M
print(ans)
0