Submission #616136

---

Source Code Expand

{-# OPTIONS_GHC -O2 #-}
{-# LANGUAGE BangPatterns #-}
module Main where
import           Control.Applicative
import           Control.Monad
import qualified Data.ByteString       as BS
import qualified Data.ByteString.Char8 as BS
import           Data.Function         (fix)
import           Data.Maybe            (fromJust)

import qualified Control.Monad.ST      as ST
import qualified Data.Vector           as V
import qualified Data.Vector.Generic   as GV
import qualified Data.Vector.Mutable   as MV
import Data.List (nub)

main :: IO ()
main = do
  n:_ <- getInts
  print $ solve n


solve :: Int -> Int
solve n = s1 - s2 
  where
    nico = [25,2525,252525,25252525]
    s1 = sum $ map (div n) nico
    s2 = sum $ map (div n) $ nub [x * y `div` gcd x y|x<- nico,y <- nico,x /= y]

  
-----------------

constructN' :: Int -> (V.Vector a -> a) -> V.Vector a
constructN' !n fun = ST.runST $ do
    v  <- MV.new n
    v' <- V.unsafeFreeze v
    flip fix 0 $ \loop !len -> do
      if len == n
        then return v'
        else do
        let v'' = V.unsafeTake len v'
            !x = fun v''
        GV.elemseq v'' x $ MV.write v len x
        loop (len+1)


getInts :: IO [Int]
getInts = fmap readInts BS.getLine

readInts :: BS.ByteString -> [Int]
readInts = map (fst . fromJust . BS.readInt). BS.words

Submission Info

Judge Result