Submission #33872

---

Source Code Expand

import Control.Applicative
import qualified Data.ByteString.Char8 as B8
import qualified Data.Vector as V
import Data.Maybe

main :: IO ()
main = do
  [n,l] <- map read . words <$> getLine
  xs <- V.replicateM l $ B8.getLine
  y <- fromJust . B8.elemIndex 'o' <$> B8.getLine

  let snake (-1) yy = yy `div` 2 + 1
      snake ll yy
        | yy < (n - 1) * 2 && B8.index x (yy + 1) == '-' = snake ll' (yy + 2)
        | yy > 0 && B8.index x (yy - 1) == '-' = snake ll' (yy - 2)
        | otherwise = snake ll' yy
        where ll' = pred ll
              x = xs V.! ll
  print $ snake (pred l) y

Submission Info

Judge Result