結果

問題 No.407 鴨等素数間隔列の数え上げ
コンテスト
ユーザー pekempey
提出日時 2016-08-11 03:21:56
言語 Haskell
(9.10.1)
結果
AC  
実行時間 207 ms / 1,000 ms
コード長 743 bytes
コンパイル時間 1,873 ms
コンパイル使用メモリ 176,128 KB
実行使用メモリ 53,120 KB
最終ジャッジ日時 2024-11-07 10:39:24
合計ジャッジ時間 10,315 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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) [3, 5..]) $ \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])
    modifySTRef ps (2:)
    forM_ [3, 5..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