結果
問題 | No.77 レンガのピラミッド |
ユーザー |
|
提出日時 | 2020-11-05 09:34:06 |
言語 | Haskell (9.10.1) |
結果 |
AC
|
実行時間 | 2 ms / 5,000 ms |
コード長 | 1,674 bytes |
コンパイル時間 | 5,060 ms |
コンパイル使用メモリ | 227,836 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-07-22 10:50:14 |
合計ジャッジ時間 | 5,878 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 20 |
コンパイルメッセージ
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 #-} import Control.Monad import Control.Monad.State import Data.Char import Data.Coerce 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 n <- readLn :: IO Int as <- seqInput n let s = (floor :: Double -> Int) . sqrt . fromIntegral . VU.sum $ as let _ans = (0 :: Int) let _now = (1 :: Int) info <- VUM.replicate 2 (0 :: Int) rep n $ \i -> do if i < s then VUM.unsafeModify info succ _now else do item <- VUM.unsafeRead info _now when (item > 0) $ VUM.unsafeModify info pred _now now <- VUM.unsafeRead info _now VUM.unsafeModify info (+ (max 0 (as VU.! i - now))) _ans print =<< VUM.unsafeRead info _ans 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 #-} 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 #-}