結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
ユーザー 情報学生
提出日時 2019-08-23 00:21:20
言語 Haskell
(9.10.1)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 990 bytes
コンパイル時間 4,621 ms
コンパイル使用メモリ 171,776 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-13 07:32:12
合計ジャッジ時間 5,284 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

{-# LANGUAGE BangPatterns #-}
data Matrix a = Matrix !a !a !a !a

matrixMul ::  Matrix Integer -> Matrix Integer -> Integer -> Matrix Integer
matrixMul (Matrix a b c d) (Matrix a' b' c' d') n
    = Matrix ((a * a' + b * c') `mod` n) ((a * b' + b * d') `mod` n) ((c * a' + d * c') `mod` n) ((c * b' + d * d') `mod` n)

matrixPow :: Matrix Integer -> Integer -> Integer -> Matrix Integer
matrixPow _ 0 _ = Matrix 1 0 0 1
matrixPow m i n = loop m m (i - 1)
    where
        loop acc !_ 0 = acc
        loop acc m 1  = matrixMul acc m n
        loop acc m i  = case i `quotRem` 2 of
                            (j, 0) -> loop acc (matrixMul m m n) j
                            (j, _) -> loop (matrixMul acc m n) (matrixMul m m n) j

fib :: Integer -> Integer -> Integer
fib i m = let Matrix a b c d = matrixPow (Matrix 0 1 1 1) (i - 1) m
        in b

main :: IO ()
main = interact $ show . solve . map (read :: String -> Integer) . words

solve :: [Integer] -> Integer
solve [n, m] = fib n m
0