結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
ユーザー かりあげクンかりあげクン
提出日時 2020-09-08 19:51:33
言語 Haskell
(9.8.2)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,399 bytes
コンパイル時間 3,201 ms
コンパイル使用メモリ 198,616 KB
実行使用メモリ 7,176 KB
最終ジャッジ日時 2023-08-20 05:14:25
合計ジャッジ時間 4,325 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,988 KB
testcase_01 AC 2 ms
7,088 KB
testcase_02 AC 3 ms
7,088 KB
testcase_03 AC 2 ms
7,136 KB
testcase_04 AC 3 ms
7,060 KB
testcase_05 AC 2 ms
7,028 KB
testcase_06 AC 3 ms
7,176 KB
testcase_07 AC 3 ms
7,040 KB
testcase_08 AC 2 ms
7,028 KB
testcase_09 AC 3 ms
7,000 KB
testcase_10 AC 3 ms
7,020 KB
testcase_11 AC 3 ms
7,020 KB
testcase_12 AC 3 ms
7,040 KB
testcase_13 AC 3 ms
7,136 KB
testcase_14 AC 2 ms
7,008 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.6.1/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

import Data.Semigroup
import Data.Monoid
import qualified Control.Arrow         as Arrow
import qualified Data.Vector.Unboxed   as VU
import qualified Data.ByteString.Char8 as BSC8
-- https://blog.miz-ar.info/2019/01/fast-fibonacci/
data FibPair a = FibPair !a !a
  deriving (Eq, Show)

instance (Num a) => Semigroup (FibPair a) where
  FibPair a b <> FibPair a' b'
    = FibPair (a * b' + (b - a) * a') (a * a' + b * b')
  stimes = stimesMonoid

instance (Num a) => Monoid (FibPair a) where
  mempty = FibPair 0 1

fibOne :: (Num a) => FibPair a
fibOne = FibPair 1 1

fibPair :: Int -> FibPair Integer
fibPair i = stimesMonoid i fibOne

fastDoubling :: Int -> Int -> (Integer, Integer)
fastDoubling 0 mo = (0, 1)
fastDoubling 1 mo = (1, 1)
fastDoubling i mo
  = let (a, b) = fastDoubling (i `quot` 2) mo
        moo    = fromIntegral mo
    in if even i
       then (a * (2 * b - a) `mod` moo, (a * a + b * b) `mod` moo)
       else ((a * a + b * b) `mod` moo, b * (2 * a + b) `mod` moo)

fib :: Int -> Int -> Integer
fib i mo = case fastDoubling (i - 1) mo of
          (a, _) -> a

main :: IO ()
main = do
  (n, mo) <- parse2
  print $ fib n mo

type Parser a = BSC8.ByteString -> Maybe (a, BSC8.ByteString)
parseInt :: Parser Int
parseInt = fmap (Arrow.second BSC8.tail) . BSC8.readInt
parse2 :: IO (Int, Int)
parse2 = (\vec -> (vec VU.! 0, vec VU.! 1)) . VU.unfoldrN 2 parseInt <$> BSC8.getLine
0