findprime

1. 문제

2. 입출력

3. 입출력 예시

4. 문제 설계

1. 주어진 문제의 크기(N)가 100이므로 각각에 대해 제곱근 까지 직접 나누어 봄으로써 소수인지 판별
2. O(N * N ^ 1/2) 시간복잡도로 문제를 해결할 수 있음

5. 전체 코드

//
//  main.swift
//  FindPrime
//
//  Created by 이영재 on 2022/03/08.
//MARK: - findprime

//MARK: - Framework
import Foundation

//MARK: - Function
func isPrimeNumber(_ number: Int) -> Bool {
    var isPrimeNumber: Bool = true
    
    if number == 1 {
        isPrimeNumber = false
    } else {
        var primeNumber: Int = 2
        while primeNumber * primeNumber <= number {
            if number % primeNumber == 0 {
                isPrimeNumber = false
                break
            }
            
            primeNumber += 1
        }
    }
    
    return isPrimeNumber
}

func solution() -> Void {
    //MARK: - Input
    guard let N: Int = Int(readLine() ?? "0") else { return }
    var primeNumberCount: Int = 0
            
    //MARK: - Process
    for _ in 0..<N {
        guard let number: Int = Int(readLine() ?? "0") else { return }
        primeNumberCount += isPrimeNumber(number) ? 1 : 0
    }
    
    //MARK: - Output
    print(primeNumberCount)
}
solution()

 

전체코드는 여기에서 확인할 수 있습니다.