백준 4948번: 베르트랑 공준(Swift)

https://www.acmicpc.net/problem/4948

 

4948번: 베르트랑 공준

베르트랑 공준은 임의의 자연수 n에 대하여, n보다 크고, 2n보다 작거나 같은 소수는 적어도 하나 존재한다는 내용을 담고 있다. 이 명제는 조제프 베르트랑이 1845년에 추측했고, 파프누티 체비쇼

www.acmicpc.net

바로 전에 풀었던 소수찾기 문제에 약간 손을 보면 된다. 이 문제 역시 에라토스테네스의 체를 사용하여 시간을 단축시키 풀 수 있다. 

import Foundation

while true{
    var sum = 0
    var input = Int(readLine()!)!
    var n = 2*input
    var numList = [Int]()
    var boolList = Array(repeating: false, count: n + 1)
    for i in 2...n {
        if !boolList[i] {
            numList.append(i)
            for j in stride(from: i * 2, through: n, by: i) {
                boolList[j] = true
            }
        }
    }
    for prime in numList {
        if prime > input{
            sum += 1
        }
    }
    print(sum)
}