結果

問題 No.99 ジャンピング駒
ユーザー こまるこまる
提出日時 2020-11-04 22:05:10
言語 Haskell
(9.8.2)
結果
AC  
実行時間 22 ms / 5,000 ms
コード長 1,361 bytes
コンパイル時間 2,603 ms
コンパイル使用メモリ 198,740 KB
実行使用メモリ 17,080 KB
最終ジャッジ日時 2023-09-29 16:20:20
合計ジャッジ時間 4,179 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 22 ms
16,960 KB
testcase_01 AC 22 ms
16,964 KB
testcase_02 AC 22 ms
16,912 KB
testcase_03 AC 22 ms
16,880 KB
testcase_04 AC 22 ms
17,004 KB
testcase_05 AC 22 ms
17,080 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.Monad.State
import           Data.Char
import           Data.Coerce
import           Data.IORef
import           Data.Word
import qualified Data.ByteString                   as BS
import qualified Data.ByteString.Char8             as BSC8
import qualified Data.Vector.Fusion.Stream.Monadic as VFSM
import qualified Data.Vector.Unboxed               as VU

main :: IO ()
main = do
  n  <- readLn :: IO Int
  xs <- seqInput n
  a  <- newIORef (0 :: Int)
  rep n $ \i -> if odd (xs VU.! i) then modifyIORef' a succ else modifyIORef' a pred
  print . abs =<< readIORef a

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 0 n)
{-# INLINE rep #-}

type CParser a = StateT BSC8.ByteString Maybe a
runCParser :: CParser a -> BSC8.ByteString -> Maybe (a, BSC8.ByteString)
runCParser = runStateT
{-# INLINE runCParser #-}
int :: CParser Int
int = coerce $ BSC8.readInt . BSC8.dropWhile isSpace
{-# INLINE int #-}
seqInput :: Int -> IO (VU.Vector Int)
seqInput n = VU.unfoldrN n (runCParser int) <$> BSC8.getLine
{-# INLINE seqInput #-}
0