백준 9421번 소수상근수

문제 링크입니다: https://www.acmicpc.net/problem/9421

에라토스테네스의 체를 이용해 쉽게 풀 수 있는 문제였습니다.

소수를 모두 구하고 N 이하인 숫자에 대해 소수상근수인지 func 함수를 통해 판별하면 되는 문제였습니다.

#include <iostream>

#include <vector>

#include <map>

using namespace std;

 

const int MAX = 1000000 + 1;

 

int minFactor[MAX];

vector<int> prime;

 

void eratosthenes(void)

{

        minFactor[0] = minFactor[1] = -1;

        for (int i = 2; i < MAX; i++)

                 minFactor[i] = i;

 

        for (int i = 2; i*i < MAX; i++)

                 if (minFactor[i] == i)

                         for (int j = i * i; j < MAX; j += i)

                                 if (minFactor[j] == j)

                                          minFactor[j] = i;

 

        for (int i = 2; i < MAX; i++)

                 if (minFactor[i] == i)

                         prime.push_back(i);

}

 

bool func(int N)

{

        map<int, bool> visited;

        int temp = N;

        while (1)

        {

                 int result = 0;

                 //각 자리수를 더해보고

                 while (temp)

                 {

                         result += (temp % 10)*(temp % 10);

                         temp /= 10;

                 }

                 //1이 나오면 소수상근수

                 if (result == 1)

                         return true;

                 //해당 숫자가 나오지 않았었다면 표시하고

                 if (!visited.count(result))

                         visited[result] = true;

                 //이미 나온 숫자라면 소수상근수가 아닙니다

                 else

                         return false;

 

                 temp = result;

        }

}

 

int main(void)

{

        ios_base::sync_with_stdio(0);

        cin.tie(0);

        int N;

        cin >> N;

 

        eratosthenes();

        for (int i = 0; i < prime.size(); i++)

                 if (prime[i] > N)

                         break;

                 else

                         if (func(prime[i]))

                                 cout << prime[i] << "\n";

       

        return 0;

}

개발환경:Visual Studio 2017

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~