結果

問題 No.37 遊園地のアトラクション
ユーザー こまるこまる
提出日時 2020-10-21 15:28:23
言語 Haskell
(9.8.2)
結果
WA  
実行時間 -
コード長 2,282 bytes
コンパイル時間 4,666 ms
コンパイル使用メモリ 251,136 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-21 08:55:02
合計ジャッジ時間 6,080 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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.Monad
import           Control.Monad.Fix
import           Control.Monad.State
import           Data.Char
import           Data.Coerce
import           Data.IORef
import qualified Data.ByteString.Char8             as BSC8
import qualified Data.Vector.Fusion.Stream.Monadic as VFSM
import qualified Data.Vector.Unboxed               as VU
import qualified Data.Vector.Unboxed.Mutable       as VUM

main :: IO ()
main = do
  t  <- readLn :: IO Int
  m  <- readLn :: IO Int
  cs <- seqInput m
  vs <- seqInput m
  dp <- VUM.replicate 10010 (0 :: Int)
  rep m $ \i -> rehf (vs VU.! i) $ \j -> rev t (cs VU.! i) $ \k -> do
    item <- VUM.unsafeRead dp (k - cs VU.! i)
    VUM.unsafeModify dp (max (item + j)) k
  print =<< VUM.unsafeRead dp t

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

streamHalf :: Monad m => Int -> Int -> VFSM.Stream m Int
streamHalf !l !r = VFSM.Stream step r
  where
    step x
      | x > l     = return $ VFSM.Yield x (x `div` 2)
      | otherwise = return $ VFSM.Done
    {-# INLINE [0] step #-}
{-# INLINE [1] streamHalf #-}

rehf :: Monad m => Int -> (Int -> m ()) -> m ()
rehf n = flip VFSM.mapM_ (streamHalf 0 n)
{-# INLINE rehf #-}

streamR :: Monad m => Int -> Int -> VFSM.Stream m Int
streamR !l !r = VFSM.Stream step (r - 1)
  where
    step x
      | x >= l    = return $ VFSM.Yield x (x - 1)
      | otherwise = return $ VFSM.Done
    {-# INLINE [0] step #-}
{-# INLINE [1] streamR #-}

rev :: Monad m => Int -> Int -> (Int -> m ()) -> m ()
rev r l = flip VFSM.mapM_ (streamR l r)
{-# INLINE rev #-}

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