[Programmers, C++]#Level2_소수 찾기, 백트래킹, 순열 백트래킹, set, 소수, set 컨테이너

#1. 문제

프로그래머스

 

---

 

#2. 풀이

 

1. 백트래킹

[알고리즘]#6_백트래킹

백트래킹은 모든 후보 경로를 점진적으로 탐색해 최적 해를 찾는 알고리즘 기법입니다. 백트래킹은 현재 경로가 문제의 최적 해를 찾을 수 없다고 판단되면, 이전 단계로 돌아가 다른 후보 경로에 대하여 탐색을 진행합니다. 일반적으로, 백트래킹은 재귀 DFS를 통해 구현합니다.

 

2. set 컨테이너

[Basic C++] #29_Set, STL 컨테이너

set 컨테이너는 C++의 STL에서 제공하는 연관 컨테이너로, 키를 중복 없이 균형 이진 트리에 저장하는 컨테이너입니다. set 컨테이너는 키의 중복을 허용하지 않고, 내부적으로 항상 정렬된 상태를 유지합니다. set 컨테이너의 접근/삽입/삭제 작업의 시간 복잡도는 O(log n)입니다.

 

3. 소수

bool IsPrime(int num)
{
    if(num <= 1) return false;
    if(num == 2) return true;
    if(num%2 == 0) return false;

    for(int i=3; i*i<=num; i+=2)
    {
        if(num%i == 0) return false;
    }

    return true;
}

위 함수는 인자로 전달받은 수의 소수 여부를 판단하는 함수입니다. 외워 두면 좋을 듯?

 

---

 

#3. 코드

// 정답 코드/*
    @링크: https://school.programmers.co.kr/learn/courses/30/lessons/42839
    @문제: 한 자리 숫자들을 조합하여 만들 수 있는 소수의 개수 구하기
    @설명
        1. 순열 백트래킹(순서 나열)
        2. 빈 문자열 체크
        3. set 컨테이너 활용
*/

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

set<int> primes;
vector<bool> visited;

bool IsPrime(int num)
{
    if(num <= 1) return false;
    if(num == 2) return true;
    if(num%2 == 0) return false;

    for(int i=3; i*i<=num; i+=2)
    {
        if(num%i == 0) return false;
    }

    return true;
}

void dfs(string cur, const string& numbers)
{
    if (!cur.empty())  // 빈 문자열 체크
    {
        try 
        {
            int num = stoi(cur);
            if (primes.find(num) == end(primes) && IsPrime(num))
            {
                primes.insert(num);
            }
        }
        catch (const std::exception& e)
        {
            // 변환 실패 또는 오버플로우 처리
            return;
        }
    }

    if(cur.length() == numbers.length()) return;

    for(int i=0; i<numbers.length(); ++i)
    {
        if(!visited[i])
        {
            visited[i] = true;
            dfs(cur + numbers[i], numbers);
            visited[i] = false;
        }
    }
}

int solution(string numbers) {
    int answer = 0;

    //@방문 여부 초기화
    visited.resize((int)numbers.size(), false);

    //@백트래킹
    string str = "";
    dfs(str, numbers);

    return primes.size();
}

 

---