백준 1978번 소수찾기 C/C++

1978번: 소수 찾기

코드1

#include <iostream>

int main()
{
	int count = 0;
	int n, num;
	int distinction = 0;
	std::cin >> n;

	for (int i = 0; i < n; i++) {
		std::cin >> num;
		if (num != 1) {
			for (int i = 1; i < num + 1; i++) {
				if (num % i == 0) {
					distinction += 1;
					if (distinction > 2) break;
				}
			}
		}
		if (distinction == 2) ++count;
		distinction = 0;
	}

	std::cout << count;

	return 0;
}

distinction 변수가 반복문이 끝났을 때 2이면 1과 숫자 본인자신으로만 나눠진 경우라 할 수 있습니다. 그래서 이 경우에 count를 증가시켜서 소수 개수를 구하였습니다.