結果

問題 No.407 鴨等素数間隔列の数え上げ
ユーザー pekempey
提出日時 2016-08-11 03:18:38
言語 Haskell
(9.10.1)
結果
AC  
実行時間 244 ms / 1,000 ms
コード長 713 bytes
コンパイル時間 5,580 ms
コンパイル使用メモリ 175,872 KB
実行使用メモリ 62,592 KB
最終ジャッジ日時 2024-11-07 10:38:59
合計ジャッジ時間 15,842 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

import Control.Applicative
import Control.Monad
import Control.Monad.ST
import Data.Array.ST
import Data.STRef

primes :: Int -> [Int]
primes n = runST $ do
    a <- newArray (0, n) True :: ST s (STUArray s Int Bool)
    forM_ (takeWhile (\i -> i * i <= n) [2..]) $ \i -> do
        v <- readArray a i
        when v $ forM_ [i * i, i * i + i .. n] $ \j -> writeArray a j False
    ps <- newSTRef [] :: ST s (STRef s [Int])
    forM_ [2..n] $ \i -> do
        v <- readArray a i
        when v $ modifySTRef ps (i:)
    readSTRef ps

solve n l = sum . map f $ primes (10 ^ 7)
    where f p = max 0 (l - p * (n - 1) + 1)
    
main = do
    [n, l] <- map read . words <$> getLine :: IO [Int]
    print $ solve n l
0