結果
問題 | No.314 ケンケンパ |
ユーザー |
|
提出日時 | 2020-10-13 17:43:35 |
言語 | Haskell (9.10.1) |
結果 |
AC
|
実行時間 | 16 ms / 1,000 ms |
コード長 | 975 bytes |
コンパイル時間 | 6,851 ms |
コンパイル使用メモリ | 173,676 KB |
実行使用メモリ | 11,884 KB |
最終ジャッジ日時 | 2024-07-20 18:37:22 |
合計ジャッジ時間 | 7,876 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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
ソースコード
{-# LANGUAGE BangPatterns #-} {-# LANGUAGE NumericUnderscores #-} import qualified Data.Vector.Fusion.Stream.Monadic as VFSM import qualified Data.Vector.Unboxed.Mutable as VUM stream :: Monad m => Int -> Int -> VFSM.Stream m Int stream !l !r = VFSM.Stream step l where step x | x < r = return $ VFSM.Yield x (x + 1) | otherwise = return $ VFSM.Done {-# INLINE [0] step #-} {-# INLINE [1] stream #-} rep :: Monad m => Int -> (Int -> m ()) -> m () rep n = flip VFSM.mapM_ (stream 3 n) {-# INLINE rep #-} modulo :: Int modulo = 1_000_000_007 {-# INLINE modulo #-} main :: IO () main = do n <- readLn :: IO Int dp <- VUM.unsafeNew (n + 1) VUM.unsafeWrite dp 0 (1 :: Int) VUM.unsafeWrite dp 1 (2 :: Int) VUM.unsafeWrite dp 2 (2 :: Int) rep n $ \i -> do item1 <- VUM.unsafeRead dp (i - 2) item2 <- VUM.unsafeRead dp (i - 3) VUM.unsafeWrite dp i (mod (item1 + item2) modulo) print =<< VUM.unsafeRead dp (n - 1)