[백준 2023 - C++] 신기한 소수

https://www.acmicpc.net/problem/2023

2023번: 신기한 소수

문제

수빈이가 세상에서 가장 좋아하는 것은 소수이고, 취미는 소수를 가지고 노는 것이다. 요즘 수빈이가 가장 관심있어 하는 소수는 7331이다.

7331은 소수인데, 신기하게도 733도 소수이고, 73도 소수이고, 7도 소수이다. 즉, 왼쪽부터 1자리, 2자리, 3자리, 4자리 수 모두 소수이다! 수빈이는 이런 숫자를 신기한 소수라고 이름 붙였다.

수빈이는 N자리의 숫자 중에서 어떤 수들이 신기한 소수인지 궁금해졌다. N이 주어졌을 때, 수빈이를 위해 N자리 신기한 소수를 모두 찾아보자.

 

입력

첫째 줄에 N(1 ≤ N ≤ 8)이 주어진다.

 

출력

N자리 수 중에서 신기한 소수를 오름차순으로 정렬해서 한 줄에 하나씩 출력한다.

 

풀이

왼쪽 부터 숫자를 한글자씩 더 채워가면서 소수인지 판별하면 된다.

이때 한자리인 소수는 2, 3, 5, 7 뿐이므로 처음에 시작을 이것으로 한다.

각각의 시작 숫자부터 N자리가 되기까지 재귀함수를 이용하여 한자리씩 추가하면서 소수인지 여부를 확인하고,

중간에 소수가 아닌 경우가 생긴다면 더 이상 그 뒤에 숫자를 추가 하지 않는다.

이렇게 소수인 상태로 N자리 수가 된다면, 벡터형 변수에 push해준다.

그리고 모두 탐색을 완료한 뒤 정렬하여 출력해주면 끝난다.

 

#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int n;
vector<long long> sol;

int check(long long n){
    for(int i=2; i*i <= n; i++){
        if(n % i ==0)
            return 0;
    }
    return 1;
}

void dfs (int now, long long num){
    int flag;
    long long temp;
    
    if(now == n){
        sol.push_back(num);
        return;
    }
    
    for(int i=0; i<10; i++){
        temp = num * 10;
        temp +=i;
        flag = check(temp);
        if(flag == 0)
            continue;
        
        dfs(now+1, temp);
    }
}

int main (void){
    
    long long num;
    
    scanf("%d", &n);
    
    
    num = 2;
    dfs(1, num);
    num = 3;
    dfs(1, num);
    num = 5;
    dfs(1, num);
    num = 7;
    dfs(1, num);
    
    sort(sol.begin(), sol.end());
    
    for(int i=0; i<sol.size(); i++)
        printf("%lld\n", sol[i]);
}