[프로그래머스] 소수 찾기 / C++

문제 설명

한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.

각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.

 

제한사항

• numbers는 길이 1 이상 7 이하인 문자열입니다.
• numbers는 0~9까지 숫자만으로 이루어져 있습니다.
• 013은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.

입출력 예

numbers	return
17	3
011	2

풀이

순열을 이용하면 손쉽게 풀 수 있는 문제였다. 세상에는 정말 똑똑한 사람들이 너무 많고..나는 그 사람들 발끝도 못 따라가고...알고리즘 공부하면서 느는건 자괴감뿐이다...........

#include <string>
#include <vector>
#include <algorithm>
#include <iostream>

using namespace std;

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

int solution(string numbers) {
    int answer = 0;
    vector<int> res;
    
    sort(numbers.begin(), numbers.end());
    
    do {
        for(int i=1; i<=numbers.size(); i++) {
            string temp = numbers.substr(0, i);
        if(find_prime(stoi(temp)) && find(res.begin(), res.end(), stoi(temp)) == res.end())
            res.push_back(stoi(temp));
        } 
    }while(next_permutation(numbers.begin(), numbers.end()));
    
    answer = res.size();
    
    return answer;
}