Math(1) Sieve of Eratosthenes(에라토스테네스의 체)

 

정의 : https://ko.wikipedia.org/wiki/%EC%97%90%EB%9D%BC%ED%86%A0%EC%8A%A4%ED%85%8C%EB%84%A4%EC%8A%A4%EC%9D%98_%EC%B2%B4

에라토스테네스의 체 - 위키백과, 우리 모두의 백과사전

 

[ Code ]

 

특정 수의 소수를 구하는 방식

fun sieveOfEratosthenes(n: Int): List<Int> {
    val isPrime = BooleanArray(n + 1) { true }
    val primes = mutableListOf<Int>()

    for (p in 2..n) {
        if (isPrime[p]) {
            primes.add(p)
            for (i in p * p..n step p) { isPrime[i] = false }
        }
    }

    return primes
}

 

특정 수들의 범위 안의 소수를 구하는 방식

fun sieveOfEratosthenes(start: Int, end: Int): List<Int> {
    val primes = mutableListOf<Int>()
    val isPrime = BooleanArray(end - start + 1) { true }

    val sqrtEnd = sqrt(end.toDouble()).toInt()
    for (p in 2..sqrtEnd) {
        for (i in maxOf(p * p, (start + p - 1) / p * p) until end + 1 step p) { isPrime[i - start] = false }
    }

    for (i in isPrime.indices) { if (isPrime[i]) primes.add(start + i) }

    return primes
}