結果

問題 No.106 素数が嫌い!2
ユーザー はむ吉🐹
提出日時 2015-10-29 22:37:19
言語 Haskell
(9.10.1)
結果
AC  
実行時間 170 ms / 5,000 ms
コード長 881 bytes
コンパイル時間 7,298 ms
コンパイル使用メモリ 173,312 KB
実行使用メモリ 30,840 KB
最終ジャッジ日時 2024-09-13 04:41:37
合計ジャッジ時間 9,289 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 13
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 Data.Array.Unboxed (UArray, accumArray, elems)
import Control.Applicative ((<$>))

-- | Generates the sequence of prime numbers.
-- Based on <http://qiita.com/little_Haskeller/items/614a3ae20a517c19bb1f this article> .
primes2 :: Integral a => [a]
primes2 = map fromIntegral primes2'
    where
        primes2' = [2, 3, 5] ++ sieve2 5 7 (drop 2 primes2')
        sieve2 m s (p : ps) = [n | n <- ns, gcd m n == 1] ++ sieve2 (m * p) (p * p) ps
            where ns = [x + y | x <- [s, s + 6 .. p * p - 2], y <- [0, 4]]

primesTo :: Int -> [Int]
primesTo n = takeWhile (<= n) primes2

count :: Int -> Int -> Int
count n k = length $ filter (>= k) $ elems tb
    where
        tb = accumArray (+) 0 (2, n) [(p * i, 1) |
            p <- primesTo n, i <- [1 .. div n p]] :: UArray Int Int

main :: IO ()
main = do
    [n, k] <- map read . words <$> getLine
    print $ count n k
0