한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.
각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.
numbers는 길이 1 이상 7 이하인 문자열입니다.
numbers는 0~9까지 숫자만으로 이루어져 있습니다.
"013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
bool is_prime(int num)
{
if(num==1 || num==0)
return false;
for(int i=2; i*i <= num; i++)
if(num%i ==0)
return false;
return true;
}
int solution(string numbers) {
vector<char> num;
vector<int> temp;
for (int i = 0; i<numbers.size(); i++)
{
num.push_back(numbers[i]);
}
sort(num.begin(), num.end());
do
{
string str = "";
for (int i = 0; i<num.size(); i++)
{
str += num[i];
temp.push_back(stoi(str));
}
} while (next_permutation(num.begin(), num.end()));
sort(temp.begin(), temp.end());
temp.erase(unique(temp.begin(), temp.end()), temp.end());
//중복제거
int answer = 0;
for (auto x : temp)
{
if (is_prime(x))
{
answer++;
}
}
return answer;
}