완전탐색 - 소수찾기

문제 설명

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

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

제한사항

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

입출력 예numbersreturn

"17"	3
"011"	2

입출력 예 설명

예제 #1
[1, 7]으로는 소수 [7, 17, 71]를 만들 수 있습니다.

예제 #2
[0, 1, 1]으로는 소수 [11, 101]를 만들 수 있습니다.

• 11과 011은 같은 숫자로 취급합니다.

 

문제풀이

이 문제는 깊이탐색으로 문자에서 for문으로 하나씩 뽑고 뽑은 위치를 체크하여 여러 경우의 수를 모두 만든다.

여러 경우의 수를 재귀함수로 만들면서, set에 집어넣어 중복을 방지한다.

그리고 해당 set에 저장된 숫자들을 에라토스테네스의 체를 이용하여 소수를 판별한다.

 

#include <string>
#include <vector>
#include <set>
#include <cmath>
using namespace std;

bool IsCheck(int iNums) // 소수인지 판별하는 함수입니다.
{
    if(iNums<2)
        return false;
    
    for(int i=2;i<=sqrt(iNums);++i)
    {
        if(iNums%i==0)
            return false;
    }
    return true;
}

void dfs(string& strOrigin, string strCheck, vector<bool> vecVisited, int iDepth, set<int>& sets)
{
    if(iDepth<=0) // 끝까지 탐색 했으면 함수를 끝냅니다.
        return;
    for(int i=0;i<vecVisited.size();++i)
    {
        if(vecVisited[i]) // 이미 사용한 문자라면 사용 안해야합니다.
            continue;
        sets.emplace(stoi(strCheck+strOrigin[i]));
        vecVisited[i] = true;
        dfs(strOrigin, strCheck+strOrigin[i],vecVisited,iDepth-1,sets);
        vecVisited[i] = false;
    }
}

int solution(string numbers) {
    int answer = 0;
    set<int> sets;
    vector<bool> vecVisited;
    vecVisited.resize(numbers.size());
    
    dfs(numbers,"",vecVisited,numbers.size(),sets);
    
    for(auto& iNums : sets)
    {
        if(IsCheck(iNums))
            ++answer;
    }
    return answer;
}