문제

M이상 N이하의 소수를 모두 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 자연수 M과 N이 빈 칸을 사이에 두고 주어진다. (1 ≤ M ≤ N ≤ 1,000,000) M이상 N이하의 소수가 하나 이상 있는 입력만 주어진다.

출력

한 줄에 하나씩, 증가하는 순서대로 소수를 출력한다.

해석

1. 에라토스테네스의 체를 이용해야한다.
2. 일단 배열에 숫자를 미리 채워 놓는다.
3. 배수를 하나씩 지워나간다.

코드

시간초과 난 코드

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int M, N;

int check(int e);

int main() {
    scanf("%d %d", &M, &N);

    for(int i=M;i<=N;i++) {
        if(check(i) == 0) {
            printf("%d\n", i);
        }
    }
    return 0;
}

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

시간초과 해결한 코드

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int arr[1000001] = {0};

int M, N;

int main() {
    scanf("%d %d", &M, &N);

    for(int i=2;i<=N;i++) {
        arr[i] = i;
    }



    for(int i=2; i*i <= N ;i++) {
        if(arr[i] == 0) continue;
        
        for(int j=i*i;j<=N;j+=i) {
            arr[j] = 0;
        }
    }


    for(int i=M;i<=N;i++) {
        if(arr[i] != 0)
            printf("%d\n", arr[i]);
    }

    return 0;
}