[C++] 9421 - 소수상근수

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

9421번: 소수상근수

 

이 문제를 푸는데 알 수 없는 런타임 에러를 엄청 많이 받았다..

이유는 처음 eratos함수를 쓸 떄 int eratos()를 썼는데 return을 안써서...( 하아... )

void로 고치고 바로 AC가 떴다.

 

 

 

 

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
 
bool E[1000020];
char S[1000020];
 
int check(int n)
{
    int tmp,tmp2 = 0;
 
    tmp = n;
    if(S[n] == 'O')
        return 1;
    else if(S[n] == '#')
        return 2;
    while(tmp)
    {
        tmp2+=(tmp%10)*(tmp%10);
        tmp/=10;
    }
    if(tmp2 == 1)
    {
        S[n] = 'O';
        return 1;
    }
    S[n] = '#';
    int flag = check(tmp2);
    if(flag == 1)
    {
        S[n] = 'O';
        return 1;
    }
    else if(flag == 2)
    {
        S[n] = '#';
        return 2;
    }
    return 0;
}
 
void eratos(int n)
{
    E[0] = E[1] = 1;
    for(int i = 2; i*i <= n; ++i)
        for(int j = i*i; j <= n; j+=i)
            E[j] = 1;
}
 
int main(void)
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    memset(S, 0, sizeof(S));
    S[0] = '#';
    S[1] = 'O';
 
    int n;
    cin >> n;
    eratos(n);
    for(int i = 2; i <= n; ++i)
        check(i);
 
    for(int i = 2; i <= n; ++i)
    {
        if(E[i] == 0 && S[i] == 'O')
            cout << i << endl;
    }
 
 
    return 0;
}