結果

問題 No.407 鴨等素数間隔列の数え上げ
コンテスト
ユーザー pekempey
提出日時 2016-08-11 03:18:38
言語 Haskell
(9.14.1)
コンパイル:
ghc -rtsopts -with-rtsopts=-K1G -o a.out -O2 _filename_
実行:
./a.out
結果
AC  
実行時間 151 ms / 1,000 ms
コード長 713 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 10,616 ms
コンパイル使用メモリ 194,304 KB
実行使用メモリ 65,540 KB
最終ジャッジ日時 2026-05-07 18:52:17
合計ジャッジ時間 9,058 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.14.1/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff #
raw source code 

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