백준 1978번 : 소수 찾기

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

1. 문제

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

2. 알고리즘

소수인지 판별 하여 소수인 수를 카운트를 하여 출력

3. 코드

#include <iostream>
#include <sstream>
#include <string>
#include <algorithm>
#include <functional>
#include <vector>
#include <list>
#include <queue>
#include <map>
#include <set>
#include <stack>
using namespace std;
 
/*
* @brief - 소수 판별
*/
int prime(int n){
 
    if (n == 1) {
        return 0;
    }
    
    if (n == 2) {
        return 1;
    }
 
    for(int i=2;i<n;i++) {
        if(n % i == 0) {
            return 0;
        }
    }
    return 1;
}
 
int main() {
    std::ios::sync_with_stdio(false); cin.tie(0);
 
    int n; cin >> n;
    
    vector<int> arr(n);
    for (int i = 0; i<n; i++) {
        cin >> arr[i];
    }
 
    int sol = 0;
    for (int i = 0; i<n; i++) {
        sol += prime(arr[i]);
    }
 
    cout << sol << endl;
    return 0;
}