코딩테스트 - 백준: 소수 구하기(에라토스테네스의 체)

문제

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

입력

첫째 줄에 자연수 M과 N이 빈 칸을 사이에 두고 주어진다. (1 ≤ M ≤ N ≤ 1,000,000) M이상 N이하의 소수가 하나 이상 있는 입력만 주어진다.

출력

한 줄에 하나씩, 증가하는 순서대로 소수를 출력한다.

 

예제 입력

3 16

예제 출력

3

5

7

11

13

 

 

나의 답안:

var input = readLine()!.components(separatedBy: " ").map{Int(String($0))!} //입력 받기
var settingArray = Array(repeating: 0, count: input.last! + 1) //0으로 된 빈 배열 선언

for i in input.first!...input.last! { //3~16이면 범위 내에 있는 수들은 배열에 집어넣기
    settingArray[i] = i				  //범위 내에 없으면 0이 됨
}

for i in 2...input.last! { //2부터 체 거르기 시작
    for j in stride(from:i+i, through:input.last! ,by: i) { //i가 2이면 4,6,8,10..이 0으로 됨(소수가 아닌것들)
        settingArray[j] = 0
    }
}

settingArray[0] = 0 //0은 소수가 아님
settingArray[1] = 0 //1도 소수가 아님

for i in input.first!...input.last! {
    if settingArray[i] != 0 { 	//배열에 있는 값이 0이 아닌 수들이 소수임
        print(settingArray[i])	//소수 출력
    }
}