[프로그래머스] 완전탐색 - 소수 찾기

문제 설명

 

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

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

 

제한사항

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

입출력 예

numbers               return

17	3
011	2

 

입출력 예 설명

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

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

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

---

해결 알고리즘

 

1. 각 숫자들의 순열을 구한다.

2. 소수 판별하기 

 

#include <string>
#include <vector>
#include <set>

using namespace std;

using Primes = set<int, greater<int>>;

bool isPrime(int number, Primes& primes) {
    if (number <= 1) {
        return false;
    }
    if (primes.find(number) != primes.end()) {
        return true;
    }
    for (int i = 2; i < number; ++i){
        if ((number % i) ==0) {
            return false;
        }
    }
    primes.insert(number);
    return true;
}

void dfs(const string& numbers, const string& currentNumber,int index, int r, vector<bool>& visited, Primes& primes) {
    if (currentNumber.size() == r) {
        const int number = stoi(currentNumber);
        if (isPrime(number, primes)) {
            return;
        }
    }

    const int numberCount = numbers.size();
    for (int i = 0; i < numberCount; ++i) {
        if (!visited[i]) {
            visited[i] = true;

            const string newNumber = currentNumber + numbers[i];
            dfs(numbers, newNumber, i, r, visited, primes);

            visited[i] = false;
        }
    }
}

int solution(string numbers) {
    const int numberCount = numbers.size();
    Primes primes;
    vector<bool> visited(numberCount, false);

    for (int r = 1; r <= numberCount; ++r) {
        dfs(numbers, "", 0, r, visited, primes);
    }

    const int answer = primes.size();
    return answer;
}