結果

問題 No.407 鴨等素数間隔列の数え上げ
ユーザー pekempey
提出日時 2016-08-11 03:35:13
言語 Haskell
(9.10.1)
結果
WA  
実行時間 -
コード長 660 bytes
コンパイル時間 1,942 ms
コンパイル使用メモリ 174,336 KB
実行使用メモリ 32,768 KB
最終ジャッジ日時 2024-11-07 10:39:28
合計ジャッジ時間 3,338 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 2
other AC * 23 WA * 8
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 = l - p * (n - 1) + 1
    
main = do
    [n, l] <- map read . words <$> getLine :: IO [Int]
    print $ solve n l
0