[백준 온라인 저지] 1978. 소수 찾기

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

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

 

1)  첫 코드(2023/02/20)

import java.io.*;
class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.valueOf(br.readLine());
        String[] input = br.readLine().split(" ");
        int answer = 0;
        for(int i=0 ; i<n ; i++){
            int a = Integer.valueOf(input[i]);
            boolean isPrime = a>1;
            for(int j=2 ; j<a ; j++){
                if(a%j==0){
                    isPrime=false; break;
                }
            }
            if(isPrime) answer++;
        }
        System.out.print(answer);
    }
}

- 14120KB, 124ms

 

2) 에라토스테네스의 체를 이용한 코드(2023/07/01)

import java.util.*;
import java.io.*;
class Main {
    public static void main(String[] args) throws IOException {
        // 입력 받기
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.valueOf(br.readLine()); // 100 이하의 자연수
        StringTokenizer token = new StringTokenizer(br.readLine());
        int[] num = new int[n]; // 1000 이하의 자연수
        for(int i=0 ; i<n ; i++)
            num[i] = Integer.valueOf(token.nextToken());
        
        // 에라토스테네스의 체를 이용해 1000 이하의 자연수 중 소수 체크
        boolean[] prime = new boolean[1001];
        Arrays.fill(prime, true); prime[1] = false;
        for(int i=2 ; i<=1000 ; i++){
            if(prime[i]){
                for(int j=2*i ; j<=1000 ; j+=i)
                    prime[j] = false;
            }
        }
        
        // 입력 받은 자연수가 소수이면 count++
        int count = 0;
        for(int i=0 ; i<n ; i++)
            if(prime[num[i]]) count++;
        System.out.print(count);
    }
}

- 14180KB, 120ms