[백준1978]-소수 찾기(C++)

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

1978번: 소수 찾기

소수 구하기

2보다 작으면 무조건 소수가 아니므로 false를 리턴, 그게 아닐 경우 2부터 해당 수의 제곱근으로 나눠보며 한번이라도 나누어 떨어지면 소수가 아니라고 판정하는 방식을 이용하였다. 코드는 아래와 같다. sqrt를 사용하기 위해서 #include <cmath> 를 해주어야 한다.

bool isPrime(int p) {
	if (p < 2) {
		return false;
	}
	double sq = sqrt(p);
	for (int i = 2; i <= sq; i++) {
		if (p % i == 0) {
			return false;
		}
	}
	return true;
}

전체 코드

#include <iostream>
#include <vector>
#include <cmath>
using namespace std;
vector <int> num;

bool isPrime(int p) {
	if (p < 2) {
		return false;
	}
	double sq = sqrt(p);
	for (int i = 2; i <= sq; i++) {
		if (p % i == 0) {
			return false;
		}
	}
	return true;
}

int main() {
	int N;
	int a;
	cin >> N;
	while (N--) {
		cin >> a;
		num.push_back(a);
	}
	int count = 0;
	for (int i = 0; i < num.size(); i++) {
		if (isPrime(num[i])) {
			count++;
		}
	}
	cout << count;
}