백준 1978 소수 찾기 문제풀이 (JAVA)

문제 링크

문제

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

입력

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

출력

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

풀이

에라토스테네스의 체를 사용하면 되지만, 맨 처음 프로그래밍을 배울 때 교수님이 소수 구하는 방법 중 이 방법을 알려주셨기에 그 댱시에는 다음과 같은 방법으로 해보았다.
처음에 1부다 해당 숫자까지 비교해서 만약 나누어서 떨어지는 숫자가 2개라면 (1과 자기 자신이라면) 해당 수는 소수이고, 그 소수를 box에 있는 숫자와 비교하여 만약 박스에 그 숫자가 있으면 정답 개수를 ++한다.
거의 2년이 다되어가는데, 나중에 다시 한번 풀어봐야겠다!

소스코드

import java.util.Scanner;

class Main{
	public static void main(String [] args) {	
		Scanner scan = new Scanner(System.in);
		int n = scan.nextInt();
		int[] box = new int[n];
		for(int i=0;i<n;i++) {
			box[i]=scan.nextInt();
		}
		scan.close();
		int count=0;
		int ans_count=0;
		for(int i=1;i<1000;i++) {
			for(int j=1;j<=i;j++) {
				if(i%j==0) {
					count++;
				}
			}
			if(count==2) {
				for(int k=0;k<n;k++) {
					if(box[k]==i) {
						ans_count++;
					}
				}
			}
			count=0;
		}
		System.out.println(ans_count);
	}	
}