結果

問題 No.314 ケンケンパ
ユーザー lvs7klvs7k
提出日時 2018-11-08 21:28:08
言語 Haskell
(9.8.2)
結果
AC  
実行時間 42 ms / 1,000 ms
コード長 1,304 bytes
コンパイル時間 3,726 ms
コンパイル使用メモリ 183,332 KB
実行使用メモリ 29,920 KB
最終ジャッジ日時 2023-08-13 04:25:37
合計ジャッジ時間 5,911 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
29,828 KB
testcase_01 AC 3 ms
6,836 KB
testcase_02 AC 2 ms
6,972 KB
testcase_03 AC 3 ms
6,908 KB
testcase_04 AC 3 ms
6,924 KB
testcase_05 AC 3 ms
6,992 KB
testcase_06 AC 2 ms
7,156 KB
testcase_07 AC 3 ms
7,012 KB
testcase_08 AC 2 ms
6,928 KB
testcase_09 AC 2 ms
7,060 KB
testcase_10 AC 3 ms
6,932 KB
testcase_11 AC 3 ms
6,976 KB
testcase_12 AC 2 ms
6,944 KB
testcase_13 AC 2 ms
6,972 KB
testcase_14 AC 3 ms
7,080 KB
testcase_15 AC 3 ms
7,176 KB
testcase_16 AC 4 ms
7,496 KB
testcase_17 AC 7 ms
10,072 KB
testcase_18 AC 22 ms
17,600 KB
testcase_19 AC 42 ms
29,920 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 # 

{-# 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