백준 1978 소수 찾기 (Java)

URL

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

1978번: 소수 찾기

문제

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

입력

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

출력

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

성공 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static boolean prime(int number) {
        if(number < 2) {
            return false;
        }

        for (int i=2; i<=number/2; i++) {
            if(number%i == 0) {
                return false;
            }
        }

        return true;
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        int testCount = Integer.parseInt(br.readLine());
        int primeCount = 0;

        String[] input = br.readLine().split(" ");

        for (int i=0; i<testCount; i++) {
            if(prime(Integer.parseInt(input[i]))) {
                primeCount += 1;
            }
        }

        System.out.println(primeCount);
    }
}