[백준] 1978번 - 소수 찾기 [Java]

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

 

[백준] 1978번 - 소수 찾기 [Java]

---

1.  아이디어

 

소수 판정에 유용한 에라토스테네스의 체를 활용하면 간단하게 해결할 수 있다.

---

2. 문제풀이

 

N의 범위가 1,000 이하의 자연수로 주어져 있으므로 1,000 이하의 소수를 판정하는 boolean 타입 배열을 구하고 이를 대조하는 방식으로 구현했다.

---

3. 코드

 

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int N = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());

        boolean[] isPrime = getPrimes(1_000);
        int ans = 0;

        while (st.hasMoreTokens()) {
            if (isPrime[Integer.parseInt(st.nextToken())]) ans++;
        }

        System.out.println(ans);
    }

    // 에라토스테네스의 체
    private static boolean[] getPrimes(int max) {
        boolean[] isPrime = new boolean[1 + max];
        Arrays.fill(isPrime, true);
        isPrime[1] = false;

        for (int i = 2; i * i <= max; i++) {
            if (isPrime[i]) {
                for (int j = i * i; j <= max; j += i) {
                    isPrime[j] = false;
                }
            }
        }

        return isPrime;
    }

}

---

4. 후기