結果

問題 No.313 π
コンテスト
ユーザー こまる
提出日時 2020-11-08 00:40:30
言語 Haskell
(9.10.1)
結果
WA  
実行時間 -
コード長 1,743 bytes
コンパイル時間 7,461 ms
コンパイル使用メモリ 217,788 KB
実行使用メモリ 8,448 KB
最終ジャッジ日時 2024-07-22 14:59:13
合計ジャッジ時間 5,053 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other WA * 32
権限があれば一括ダウンロードができます
コンパイルメッセージ
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    #-}
{-# LANGUAGE OverloadedLists #-}

import           Control.Monad
import           Control.Monad.State
import           Data.Coerce
import           Data.Word
import           Unsafe.Coerce
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
import qualified Data.Vector.Unboxed.Mutable       as VUM

piDist :: VU.Vector Int
piDist = [20104, 20063, 19892, 20011, 19874, 20199, 19898, 20163, 19956, 19841]
{-# INLINE piDist #-}

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

main :: IO ()
main = do
  bs <- (VU.map (subtract (48 :: Word8))) <$> seqInput 200002
  dp <- VUM.replicate 10 (0 :: Int)
  rep 200002 $ \i -> when (bs VU.! i /= 254) $ VUM.unsafeModify dp succ (unsafeCoerce $ bs VU.! i)
  xs <- VU.unsafeFreeze dp
  let ys = VU.zipWith (-) piDist xs
      a  = VU.findIndex (< 0) ys
      b  = VU.findIndex (> 0) ys
  putStrLn $ show a <> " " <> show b

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