{-# LANGUAGE BangPatterns #-} {-# LANGUAGE NumericUnderscores #-} import Control.Monad import Control.Monad.ST import Data.IORef import Data.STRef import qualified Data.Vector.Fusion.Stream.Monadic as VFSM main :: IO () main = do x <- readLn :: IO Int if x == 2 then putStrLn "1000000" else do res <- newIORef (0.0 :: Double) rep2 $ \a -> modifyIORef' res (+ zeta a x) print . floor . (* 1_000_000) =<< readIORef res zeta :: Int -> Int -> Double zeta a x = runST $ do res <- newSTRef (a - 1) rep (x - 1) $ \_ -> do ret <- readSTRef res when (ret <= 1_000_000_000_000) $ modifySTRef res (* a) d <- readSTRef res return (1.0 / fromIntegral d) 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 #-} rep2 :: Monad m => (Int -> m ()) -> m () rep2 = flip VFSM.mapM_ (stream 2 1_000_000)