1978번: 소수 찾기

문제 출처 : https://www.acmicpc.net/problem/1978

1978번: 소수 찾기

언어 : Kotlin

 

문제 설명 :

주어진 수 N개 중에서 소수가 몇 개인지 찾아서 출력하는 프로그램을 작성하시오.

• 첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.
• 주어진 수들 중 소수의 개수를 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
4 1 3 5 7	3

풀이 : 

import java.io.*

fun main(args: Array<String>) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    var cnt = 0

    val size = br.readLine()
    val n = br.readLine().split(" ").map { it.toInt() }

    for (i in n.indices) {
        var arr = emptyArray<Int>()

        for (j in 1 .. n[i]) {
            if (n[i] % j == 0) arr += n[i]
        }

        if (arr.size == 2) cnt++
    }

    bw.write("$cnt")
    bw.flush()
    bw.close()
}