백준 1978

문제

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

입력

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

출력

주어진 수들 중 소수의 개수를 출력한다.

예제 입력 1

4
1 3 5 7

예제 출력 1

3

풀이(kotlin)

- 에라토스테네스의 체를 이용하여 1000까지의 소수를 배열에 구하고 소수를 판별하는 풀이를 사용하였다.

private val r = System.`in`.bufferedReader()

fun main() {
    r.readLine()
    val arr = IntArray(1001)
    arr[1] = 1

    for (i in 2 until 1001) {
        for (j in i*i until 1001 step i) {
            arr[j] = 1
        }
    }

    var ans = 0
    r.readLine().split(" ").map { it.toInt() }.toIntArray().apply {
        for (i in this) if (arr[i] == 0) ans++
    }
    println(ans)
}

풀이(C99)

#include <stdio.h>

int main() {
    int N, num;
    int n = 0;
    scanf("%d", &N);
    int arr[1001] = {0};
    arr[1] = 1;
    for (int i = 2; i < 1001; i++) {
        for (int j = i * i; j < 1001; j += i) {
            arr[j] = 1;
        }
    }

    for (int i = 0; i < N; i++) {
        scanf("%d", &num);
        if (arr[num] == 0) n++;
    }

    printf("%d", n);
    return 0;
}