結果

問題 No.106 素数が嫌い！2
ユーザー ducktailducktail
提出日時 2018-07-23 13:50:09
言語 Haskell
(9.8.2)
結果
AC  
実行時間 72 ms / 5,000 ms
コード長 774 bytes
コンパイル時間 2,808 ms
コンパイル使用メモリ 185,960 KB
実行使用メモリ 25,480 KB
最終ジャッジ日時 2023-08-27 05:35:46
合計ジャッジ時間 4,303 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 32 ms
18,220 KB
testcase_01 AC 2 ms
7,000 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 3 ms
6,820 KB
testcase_04 AC 32 ms
18,312 KB
testcase_05 AC 63 ms
25,452 KB
testcase_06 AC 62 ms
25,440 KB
testcase_07 AC 61 ms
25,480 KB
testcase_08 AC 12 ms
11,988 KB
testcase_09 AC 12 ms
12,088 KB
testcase_10 AC 62 ms
25,448 KB
testcase_11 AC 32 ms
17,160 KB
testcase_12 AC 72 ms
25,228 KB
testcase_13 AC 2 ms
6,860 KB
testcase_14 AC 2 ms
6,788 KB
testcase_15 AC 3 ms
6,904 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.6.1/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