結果

問題 No.106 素数が嫌い！2
ユーザー ducktailducktail
提出日時 2018-07-23 13:50:09
言語 Haskell
(9.8.2)
結果
AC  
実行時間 84 ms / 5,000 ms
コード長 774 bytes
コンパイル時間 6,407 ms
コンパイル使用メモリ 195,840 KB
実行使用メモリ 23,680 KB
最終ジャッジ日時 2024-06-08 01:29:23
合計ジャッジ時間 6,746 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
15,744 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 32 ms
15,872 KB
testcase_05 AC 66 ms
23,552 KB
testcase_06 AC 66 ms
23,552 KB
testcase_07 AC 84 ms
23,552 KB
testcase_08 AC 8 ms
8,960 KB
testcase_09 AC 7 ms
9,216 KB
testcase_10 AC 70 ms
23,680 KB
testcase_11 AC 26 ms
14,976 KB
testcase_12 AC 79 ms
22,784 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 (when, forM_, foldM)
import Data.Vector.Unboxed.Mutable (IOVector, STVector)
import qualified Data.Vector.Unboxed.Mutable as VM

main :: IO ()
main = solve >>= print

solve :: IO Int
solve = do
  [n, k] <- map read <$> words <$> getLine :: IO [Int]
  pt <- VM.replicate (n+1) 0 :: IO (IOVector Int)
  forM_ [2 .. n] $ \i -> do
    x <- VM.read pt i
    when (x == 0) $ do
      VM.write pt i 1
      forM_ [2*i, 3*i .. n] $ \j -> do
        modify pt j (+1)
  foldM (f k pt) 0 [2 .. n]
  where f k pt c i = do
          x <- VM.read pt i
          if x >= k then return (c+1)
            else return c
  
modify :: IOVector Int -> Int -> (Int -> Int) -> IO ()
modify v i f = do
  x <- VM.read v i
  VM.write v i (f x)
0