Do Something

[프로그래머스][c++][완전탐색] 소수 찾기(dfs) 본문

프로그래머스

[프로그래머스][c++][완전탐색] 소수 찾기(dfs)

뭐라도해야겠다 2022. 1. 30. 22:25
  • 소수 찾기
문제 설명

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

각 종이 조각에 적힌 숫자가 적힌 문자열 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은 같은 숫자로 취급합니다.

출처


딱히 할 말이 없다.

#include <string>
#include <vector>
#include <iostream>
#include <math.h>
using namespace std;

bool visit[8],vvisit[10000000];
vector<int> ans;
string number,temp;
bool isprime(int num){
    if(num==1 || num == 0) return false;
    if(num == 2) return true;
    int t = sqrt(num);
    for(int i=2;i<t+1;i++){if(num % i == 0)  return false;}
    return true;
}

void dfs(int dep){
    if(dep == number.size()+1) return;
    int i = 0,tmp;
    for(auto K:number){
        if(visit[i] == false) {
            visit[i] = true;
            temp.push_back(number[i]);
            tmp = stoi(temp);
            if(isprime(tmp)){
                if(vvisit[tmp]== false){
                    vvisit[tmp] = true;
                    ans.push_back(tmp);
                }
            }
            dfs(dep+1);
            visit[i] = false;
            temp.pop_back();
        }
        i++;
    }
}

int solution(string numbers) {
    int answer = 0;
    number = numbers;
    dfs(0);
    answer = ans.size();
    return answer;
}