結果

問題 No.314 ケンケンパ
ユーザー lvs7k
提出日時 2018-11-08 21:10:19
言語 Haskell
(9.10.1)
結果
AC  
実行時間 853 ms / 1,000 ms
コード長 666 bytes
コンパイル時間 10,510 ms
コンパイル使用メモリ 187,520 KB
実行使用メモリ 254,592 KB
最終ジャッジ日時 2024-11-20 21:09:24
合計ジャッジ時間 10,799 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 #-}

import Control.Applicative
import Control.Monad
import Data.List
import Debug.Trace (traceShow)
import qualified Data.ByteString.Char8 as B

import Data.Array

inf :: Int
inf = 10 ^ 9 + 7

solve :: Int -> Int
solve n = (memo ! (n, 0) + memo ! (n, 1) + memo ! (n, 2)) `mod` inf
  where
    memo :: Array (Int, Int) Int
    memo = listArray ((1, 0), (n, 2)) [go i j | i <- [1 .. n], j <- [0 .. 2]]

    go 1 0 = 0
    go 1 1 = 1
    go 1 2 = 0
    go i 0 = memo ! (i - 1, 1) `mod` inf + memo ! (i - 1, 2) `mod` inf
    go i 1 = memo ! (i - 1, 0)
    go i 2 = memo ! (i - 1, 1)

main :: IO ()
main = do
    n <- readLn
    print $ solve n
0