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

 

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

 

 

<풀이>

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

bool prime(int x){
   
    if(x == 0 || x == 1) return 0;
   
       for(int i=2; i<x; i++){
           if(x % i == 0){
               return 0; 
           }
       }
    
   return 1; 
  
}


int solution(vector<int> nums) {
    int answer = 0;
    int res = 0;
    vector<int> v;
    
    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++){
                res+=nums[i]+nums[j]+nums[k];
                v.push_back(res);
                res = 0;
         }
      }
    }

    for(int i=0; i<v.size(); i++){
       if(prime(v[i])) answer++;
    }
    
    return answer;
}