백준 1929번: 소수 구하기 - Swift 풀이

오늘의 문제

1929번: 소수 구하기

 

내 풀이

import Foundation

let input = readLine()!.split(separator: " ").map { Int($0)! }
var M = input[0]
var N = input[1]
var arr: [Bool] = Array(repeating: true, count: N+1)
arr[0] = false
if N >= 1 { arr[1] = false }

// 1 1 일 때는 그냥 강제로 0 출력해버림
if M == 1 && N == 1 {
    print(0)
    exit(0)
}

if N > 2 {
    // 에라토스테네스의 체를 사용
    for i in 2..<Int(sqrt(Double(N))+1) {
        if arr[i] {
            for j in stride(from: i*i, to: N+1, by: i) {
                arr[j] = false
            }
        }
    }
}

for i in M...N {
    if arr[i] {
        print(i)
    }
}