[프로그래머스] 소수 만들기 : C++

https://programmers.co.kr/learn/courses/30/lessons/12977

코딩테스트 연습 - 소수 만들기

#include <vector>
#include <math.h>
using namespace std;


bool checkPrimeNumber(int pnum){
    int sqrt_pnum = (int) sqrt(pnum);
    for(int i = 2; i <= sqrt_pnum; i++) if(pnum % i == 0) return false;
    return true;
}

int solution(vector<int> nums) {
    int answer = 0;
    int pnum;
    for(int i = 0; i < nums.size() -2; i++){
        for(int j = i + 1; j < nums.size() - 1; j++){
            for(int k = j + 1; k < nums.size(); k++){
                pnum = nums[i] + nums[j] + nums[k];
                if(checkPrimeNumber(pnum)) answer++;
            }
        }
    }
    return answer;
}

벡터의 길이가 최대 50이기 때문에 중첩 for문에서 자유로울 수 있었다.

모든 경우의 수를 구하고 각 경우의 수 마다 소수인지 판별해 카운트한다.