結果

問題 No.720 行列のできるフィボナッチ数列道場 (2)
ユーザー Common LispCommon Lisp
提出日時 2024-11-13 02:06:44
言語 Common Lisp
(sbcl 2.3.8)
結果
WA  
実行時間 -
コード長 1,567 bytes
コンパイル時間 315 ms
コンパイル使用メモリ 43,624 KB
実行使用メモリ 32,020 KB
最終ジャッジ日時 2024-11-13 02:06:46
合計ジャッジ時間 1,879 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 10 ms
25,896 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 10 ms
26,028 KB
testcase_19 AC 10 ms
25,896 KB
testcase_20 AC 11 ms
27,976 KB
testcase_21 AC 10 ms
30,016 KB
testcase_22 WA -
権限があれば一括ダウンロードができます
コンパイルメッセージ
; compiling file "/home/judge/data/code/Main.lisp" (written 13 NOV 2024 02:06:44 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~%" (aref (%mod-mat-pow e (1+ n) +mod+) 2 0))))

(main)
0