1929번: 소수 구하기

문제 출처 : https://www.acmicpc.net/problem/1929

1929번: 소수 구하기

언어 : Kotlin

 

문제 설명 :

M이상 N이하의 소수를 모두 출력하는 프로그램을 작성하시오.

• 첫째 줄에 자연수 M과 N이 빈 칸을 사이에 두고 주어진다. (1 ≤ M ≤ N ≤ 1,000,000) M이상 N이하의 소수가 하나 이상 있는 입력만 주어진다.
• 한 줄에 하나씩, 증가하는 순서대로 소수를 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 256MB

 

입출력 예 :

입력	출력
3 16	3 5 7 11 13

풀이 : 

import java.io.*
import kotlin.math.sqrt

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val (n, m) = br.readLine().split(" ").map { it.toInt() }
    (n .. m).map { if (isPrime(it.toLong())) bw.appendLine("$it") }
    bw.flush()
    bw.close()
}

fun isPrime(n: Long): Boolean {
    if (n <= 1) return false
    return (2 .. sqrt(n.toDouble()).toInt()).none { n % it == 0.toLong() }
}