結果

問題 No.407 鴨等素数間隔列の数え上げ
ユーザー pekempey
提出日時 2016-08-11 03:36:03
言語 Haskell
(9.10.1)
結果
AC  
実行時間 98 ms / 1,000 ms
コード長 668 bytes
コンパイル時間 12,970 ms
コンパイル使用メモリ 177,408 KB
実行使用メモリ 32,768 KB
最終ジャッジ日時 2024-12-15 23:07:34
合計ジャッジ時間 3,506 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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)
    ps <- newSTRef [2] :: ST s (STRef s [Int])
    forM_ [3, 5 .. n] $ \i -> do
        v <- readArray a i
        when v $ do
            modifySTRef ps (i:)
            forM_ [i * i, i * i + i .. n] $ \j -> writeArray a j False
    readSTRef ps

solve n l = sum . map f $ primes ((l + 1) `div` (n - 1))
    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