소수 만들기 (feat. 프로그래머스, 12977번)

소수 만들기
https://programmers.co.kr/learn/courses/30/lessons/12977

프로그래머스

반응형

// woohyeon.kim
#include <vector>
#include <algorithm>

using namespace std;

bool is_prime(const int& n) {
	for (auto idx = 2; idx < n; ++idx)
    {
		if ((n % idx) == 0)
			return false;
	}
	return true;
}

int solution(vector<int> nums) {
    int answer = 0;
    auto N = nums.size();
    vector<bool> check(N);
    fill(check.end() - 3, check.end(), true);
    do
    {
        auto value = 0;
        for(auto idx = 0; idx < N; ++idx)
        {
            if(check[idx])
                value += nums[idx];
        }
        if(is_prime(value))
            ++answer;
    } while(next_permutation(check.begin(), check.end()));

    return answer;
}
// *&)*@*

순열을 이용하여 3개의 조합이 되는 모든 경우에 대해 소수를 체크하는 형태입니다.