結果

問題 No.106 素数が嫌い！2
ユーザー かりあげクンかりあげクン
提出日時 2020-09-25 10:06:40
言語 Haskell
(9.8.2)
結果
AC  
実行時間 37 ms / 5,000 ms
コード長 875 bytes
コンパイル時間 9,142 ms
コンパイル使用メモリ 207,616 KB
実行使用メモリ 19,328 KB
最終ジャッジ日時 2024-06-28 05:46:33
合計ジャッジ時間 10,682 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 14 ms
11,264 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 3 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 17 ms
11,520 KB
testcase_05 AC 36 ms
19,072 KB
testcase_06 AC 30 ms
19,072 KB
testcase_07 AC 35 ms
19,328 KB
testcase_08 AC 4 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 AC 34 ms
19,328 KB
testcase_11 AC 14 ms
10,604 KB
testcase_12 AC 37 ms
18,688 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 3 ms
5,376 KB
testcase_15 AC 3 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.Arrow
import           Control.Monad
import qualified Data.ByteString.Char8       as BSC8
import qualified Data.Vector.Unboxed         as VU
import qualified Data.Vector.Unboxed.Mutable as VUM

main :: IO ()
main = do
  (n, k) <- parse2
  solver n k >>= print . VU.length . VU.filter (>=k)

solver :: Int -> Int -> IO (VU.Vector Int)
solver n k = do
  table <- VU.unsafeThaw $ VU.replicate (n + 1) (0 :: Int)
  forM_ [2..n] $ \i -> do
    x <- VUM.read table i
    when (x == 0) $ do
      VUM.write table i 1
      forM_ [2 * i, 3 * i .. n] $ \j -> VUM.unsafeModify table (+1) j
  VU.unsafeFreeze table

type Parser a = BSC8.ByteString -> Maybe (a, BSC8.ByteString)
parseInt :: Parser Int
parseInt = fmap (second BSC8.tail) . BSC8.readInt
parse2 :: IO (Int, Int)
parse2 = (\vec -> (vec VU.! 0, vec VU.! 1)) . VU.unfoldrN 2 parseInt <$> BSC8.getLine
0