[Beakjoon] 1978. 소수 찾기 #B2 #Java

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

예제 입력 1

4
1 3 5 7

예제 출력 1

3

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

 

문제풀이

public class Main {
    public static void main(String[] args) throws IOException {
        // 1. 변수 선언 및 초기화
        int maxValue = 1_000;
        int[] primeArr = new int[maxValue + 1];
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        int result = 0;

        // 2. 배열 초기화
        for (int i = 2; i <= maxValue; i++) {
            primeArr[i] = 1;
        }

        // 3. 에라토스테네스의 체
        for (int i = 2; i <= maxValue; i++) {
            // 이미 제외된 숫자일 경우 건너뛰기
            if (primeArr[i] == 0) continue;

            // 아닐 경우 i를 제외한 배수들 모두 지우기
            for (int j = i * 2; j <= maxValue; j += i) {
                primeArr[j] = 0;
            }
        }

        // 4. 소수 판별
        while (st.hasMoreTokens()) {
            int number = Integer.parseInt(st.nextToken());
            if (primeArr[number] == 1) result++;
        }

        // 5. 출력
        System.out.println(result);
    }
}

• 소수를 찾는 문제는 보통 제곱근까지 구하는 방법과 에라토스테네스의 체를 이용해 해결할 수 있다.
• 에라토스테네스의 체는 다음과 같은 방법으로 소수를 판별한다.
  • 배열 요소를 모두 1로 초기화한다. 체로 걸러내기 위해 초기 모든 수를 소수로 간주하는 것이다.
  • i를 제외한 배수들을 체로 걸러낸다. 이때 이미 제외된 숫자일 경우 건너뛰면 된다.

 

참고 사이트

에라토스테네스의 체

 

[Algorithm] 에라토스테네스의 체