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