ps

소수 만들기

롯대쌍쌍바 2021. 6. 24. 14:23

 

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

bool isPrior(int num){
    if(num == 1) return false;
    for(int i = 2; i<= sqrt(num); i++){
        if(num%i == 0) return false;
    }
    return true;
}

int solution(vector<int> nums) {
    int result = 0;
    for(int i = 0; i< nums.size(); i++){
        for(int j = i+1; j < nums.size(); j++){
            for(int k = j+1; k < nums.size(); k++){
                if(isPrior(nums[i]+nums[j]+nums[k])){
                    result++;
                }
            }
        }
    }
    return result;
}