백준 1978 소수 찾기 / C++

문제 : 1978

풀이 point

1부터 n-1까지 나누어 소수인지 판단하면 된다. 단 1은 특수한 상황이다.

코드

//백준 1978, 소수 찾기
#include <iostream>


int main(){

    int N, n;
    int sum{0};

    std::cin >> N;
    
    for(int i{0}; i<N; ++i){
        std::cin >> n;
        int num{2}; 
        bool flag = n == 1 ? true : false;
        while(n > num){
            if(n%num == 0){
                flag = true;
                break;
            }
            ++num;
        }
        if(!flag) ++sum;
    }

    std::cout << sum;

    return 0;
}