4134번: 다음 소수

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

4134번: 다음 소수

언어 : Kotlin

 

문제 설명 :

정수 n(0 ≤ n ≤ 4*10⁹)가 주어졌을 때, n보다 크거나 같은 소수 중 가장 작은 소수 찾는 프로그램을 작성하시오.

• 첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다.
• 각각의 테스트 케이스에 대해서 n보다 크거나 같은 소수 중 가장 작은 소수를 한 줄에 하나씩 출력한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
3 6 20 100	7 23 101

풀이 : 

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

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

    repeat(br.readLine().toInt()) {
        var num = br.readLine().toLong()

        while (true) {
           if (isPrime(num)) {
               bw.appendLine("$num")
               break
           } else {
               num++
           }
        }
    }
    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() }
}