結果

問題 No.720 行列のできるフィボナッチ数列道場 (2)
コンテスト
ユーザー Common Lisp
提出日時 2024-11-13 02:14:55
言語 Common Lisp
(sbcl 2.5.0)
結果
AC  
実行時間 13 ms / 2,000 ms
コード長 1,579 bytes
コンパイル時間 440 ms
コンパイル使用メモリ 39,448 KB
実行使用メモリ 22,016 KB
最終ジャッジ日時 2024-11-13 02:14:57
合計ジャッジ時間 1,989 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 13 NOV 2024 02:14:55 AM):

; wrote /home/judge/data/code/Main.fasl
; compilation finished in 0:00:00.028

ソースコード

diff # 

(defconstant +mod+ 1000000007)

(defun %mod-fibonacci (n m)
  (labels ((rec (a b i p q)
                (cond ((zerop i) (mod b m))
                      ((evenp i) (rec a b (floor i 2) (mod (+ (* p p) (* q q)) m) (mod (+ (* 2 p q) (* q q)) m)))
                      (t (rec (mod (+ (* b q) (* a q) (* a p)) m) (mod (+ (* b p) (* a q)) m) (1- i) p q)))))
    (rec 1 0 n 0 1)))

(defun %mod-mat-mul (a b m)
  (let ((res (make-array (list 3 3) :element-type 'integer :initial-element 0)))
    (dotimes (i 3)
      (dotimes (j 3)
        (dotimes (k 3)
          (incf (aref res i j) (mod (* (aref a i k) (aref b k j)) m)))))
    res))

(defun %mod-mat-pow (a n m)
  (let ((res (make-array (list 3 3) :element-type 'integer)))
    (dotimes (i 3)
      (dotimes (j 3)
        (setf (aref res i j) (if (= i j) 1 0))))
    (loop while (not (zerop n))
          when (oddp n)
            do (setf res (%mod-mat-mul a res m))
          do (setf a (%mod-mat-mul a a m))
             (setf n (floor n 2)))
    res))

(defun main (&rest argv)
  (declare (ignorable argv))
  (let* ((n (read))
         (m (read))
         (a (%mod-fibonacci (1+ m) +mod+))
         (b (%mod-fibonacci m +mod+))
         (c (%mod-fibonacci m +mod+))
         (d (%mod-fibonacci (1- m) +mod+))
         (e (make-array (list 3 3) :element-type 'integer :initial-element 0)))
    (setf (aref e 0 0) a
          (aref e 0 1) b
          (aref e 1 0) c
          (aref e 1 1) d
          (aref e 2 1) 1
          (aref e 2 2) 1)
    (format t "~d~%" (mod (aref (%mod-mat-pow e (1+ n) +mod+) 2 0) +mod+))))

(main)
0