프로그래머스 12977 소수 만들기

개요

프로그래머스 12977 소수 만들기

• 프로그래머스 12977번 문제
• 프로그래머스 Lv. 1.4

C++

#include <vector>
#include <iostream>
using namespace std;

bool isPrime(int x) {
    if(x<2) return false;
    if(x<4) return true;
    if(x%2==0 || x%3==0) return false;
    for(int i=5; i*i<=x; i+=6 ) {
        if(x%i==0 || x%(i+2)==0) return false;
    }
    return true;
}

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