[백준 1978] 소수 찾기

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

1978번: 소수 찾기

#include <stdio.h>
int cnt;

void is_prime(int n)
{
	if (n == 1) return;
	for (int i = 2; ; i++) {
		if (i * i > n) break;
		if (!(n % i)) return;
	}
	cnt++;
}

int main(void)
{
	int num, n;
	scanf("%d", &num);

	for (int i = 0; i < num; i++) {
		scanf("%d", &n);
		is_prime(n);
	}
	printf("%d", cnt);
}

이 문제는 scanf로 받은 수가 소수인지 아닌지만 판별하면 되기 때문에 for문을 한개만 써도 된다. 

어차피 n이 1이거나, 약수가 존재할 때는 더는 볼 필요 없으므로 return을 하여 아예 is_prime 함수를 빠져나오도록 해줬다.

다만 i * i > n 이면 i가 n의 제곱근이 되었을 때 까지 약수가 없는 경우이므로, break를 해주어 루프를 빠져나오고 cnt를 증가시키도록 구현했다. (이 때는 소수이니까)