Submission #180150

---

Source Code Expand

{-# LANGUAGE CPP #-}
#ifndef HDEVTOOLS
{-# OPTIONS_GHC -O2 -funbox-strict-fields #-}
#endif
{-# LANGUAGE BangPatterns, ViewPatterns, OverloadedStrings #-}

import qualified Data.ByteString.Char8 as S
import qualified Data.Vector.Unboxed as U

main :: IO ()
main = do
  _ <- getLine
  print . solve . U.fromList . map readInt . S.words =<< S.getLine

solve :: U.Vector Int -> Int
solve = U.sum . U.map f
  where
    f x = x - good x

good :: Int -> Int
good n = 6 * q + r'
  where
    (q, r) = divMod n 6
    r'
      | r < 1 = -3
      | r < 3 = 1
      | otherwise = 3

readInt :: S.ByteString -> Int
readInt s = case S.readInt s of
  Just (r, "") -> r
  _ -> error $ "not an integer: " ++ show s

Submission Info

Judge Result