結果

問題 No.314 ケンケンパ
ユーザー lvs7k
提出日時 2018-11-08 21:28:08
言語 Haskell
(9.10.1)
結果
AC  
実行時間 32 ms / 1,000 ms
コード長 1,304 bytes
コンパイル時間 11,541 ms
コンパイル使用メモリ 192,512 KB
実行使用メモリ 27,404 KB
最終ジャッジ日時 2024-11-20 21:09:37
合計ジャッジ時間 3,675 ms
ジャッジサーバーID
（参考情報） 		judge1 / 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
import Control.Monad.ST
import Data.Array.ST
import Data.Array.Unboxed

inf :: Int
inf = 10 ^ 9 + 7

-- TLE
-- 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)

solve :: Int -> Int
solve n = (`mod` inf) $ dp ! (n, 0) + dp ! (n, 1) + dp ! (n, 2)
  where
    dp = runSTUArray $ do
        a <- newArray ((1, 0), (n, 2)) 0 :: ST s (STUArray s (Int, Int) Int)
        writeArray a (1, 1) 1
        forM_ [2 .. n] $ \i -> do
            v0 <- readArray a (i - 1, 0)
            v1 <- readArray a (i - 1, 1)
            v2 <- readArray a (i - 1, 2)
            writeArray a (i, 0) ((v1 + v2) `mod` inf)
            writeArray a (i, 1) v0
            writeArray a (i, 2) v1
        return a

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