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