[프로그래머스][C++, Python]소수 찾기 문제 ( 에라토스테네스의 체 )

 프로그래머스의 문제중 소수찾기를 풀어보았다.

https://programmers.co.kr/

프로그래머스

 

처음에 했던 풀이

#include <string>
#include <vector>
#include <iostream>
using namespace std;
bool IsPrimeNum(int n) {
    if ((n % 2) == 0)
        return (n == 2);
 
    for (int i = 3; i * i <= n; i++) {
        if ((n % i) == 0)
            return false;
    }
    
    return true;
}
int solution(int n) {
    int answer = 0;
    int check_num = 0;
    if (n == 2)
        return 1;
    else
        answer = 1;
 
    check_num = 3;
 
    while (check_num <= n) {
        
        if (IsPrimeNum(check_num))
            answer++;
 
        check_num = check_num + 2;// 홀수만
    }
    return answer;
}
 

처음에는 순차적으로 소수인지 모두 판별하였다(짝수를 제외하고)

 

실행 결과값은 맞았지만 효율성에서 매우 fail

 

 

그래서 에라토스테네스의 체를 사용했다

에라토스테네스의 체 (출처 위키백과)

 

에라토스테네스의 체를 이용한 C++ 풀이

#include <string>
#include <vector>
#include <cstdlib>
 
using namespace std;
 
int solution(int n) {
    int answer = 0;
 
    bool* IsPrime = new bool[n+1];
    for (int i = 2; i <= n; i++) {
        IsPrime[i] = true;
    }
    for (int i = 2; i * i <= n; i++) {
        if (IsPrime[i]) {
            for (int j = i*i; j <= n; j+=i)
                IsPrime[j] = false;
        }
    }
    for (int i = 2; i <= n; i++)
        answer += IsPrime[i];
  
 
    return answer;
}
 
 

 

에라토스테네스의 체를 이용하니까 풀이도 맞고 시간효율성도 좋아서 효율성도 통과했다.