프로그래머스 소수찾기 [Java]

문제 설명

 

1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요.

소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다.
(1은 소수가 아닙니다.)

 

 

제한 조건

 

• n은 2이상 1000000이하의 자연수입니다.

 

 

입출력 예

 

 

 

내 코드중 틀린 코드

class Solution {
    public int solution(int n) {
        int answer = 0;
        int[] array = new int[n+1];
        for (int i = 2; i <=n ; i++) {
            array[i]=i;
        }
        for (int i = 2; i <n ; i++) {
            if (array[i]==0)continue;
            for (int j = i*i; j <=n ; j+=i) {
                array[j]=0;
            }
        }
        for (int i = 0; i <= n ; i++) {
            if (array[i]>0){
                answer+=1;
            }
        }
        return answer;
    }
}

 

 

정답 코드

//1. for문을 통해 0 과 1은 소수가 아니기에 3번째 인덱스부터 숫자를 넣는다.
//2. 2부터 +2 씩 증가하며 자신의 값을 배열에서 0으로 만든다.
//3. for문으로 배열에서 0보다 큰 수를 찾아 반환한다.

class Solution {
    public int solution(int n) {
        int answer = 0;
        int[] array = new int[n+1];
        for (int i = 2; i <=n ; i++) {
            array[i]=i;
        }
        for (int i = 2; i*i <=n ; i++) {
            if (array[i]==0)continue;
            for (int j = i*i; j <=n ; j+=i) {
                array[j]=0;
            }
        }
        for (int i = 0; i <= n ; i++) {
            if (array[i]>0){
                answer+=1;
            }
        }
        return answer;
    }
}

 

 

에란토스테네스의 체를 활용하여 문제를 풀었다.

런타임 에러가 나와서 봤더니 for문에서 i의 범위를 줄여주어야 했다.

만약 n값이 121이면 제곱근인 11까지만 범위를 정해주면 된다.

그 이상은 의미 없는 반복이기 때문에 시간초과가 발생하고, 런타임 에러가 발생할 수 있다.