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

Lv1의 소수찾기 문제입니다.

아리스토테네스의 체를 활용하는 문제여서

알고리즘 공부하다가 적용해 풀어보았습니다.

---

<문제 설명>

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

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

제한 조건

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

<입출력 예>

n	result
10	4
5	3

<입출력 예 설명>

입출력 예 #1
1부터 10 사이의 소수는 [2,3,5,7] 4개가 존재하므로 4를 반환

입출력 예 #2
1부터 5 사이의 소수는 [2,3,5] 3개가 존재하므로 3를 반환

---

<내 풀이>

아리스토테네스의 체를 적용하는게 관건이다.

다른 방식의 풀이는 모두 효율성 테스트를 통과하지 못한다.

 

<코드>

class Solution {
    public int solution(int n) {
        int answer = 0; 
        boolean[] arr = new boolean[n+1];
        arr[0] = arr[1] = false;
        for(int i=2; i<=n; i+=1) {
            arr[i] = true;
        }
        for(int i=2; i*i<=n; i+=1) {
            for(int j=i*i; j<=n; j+=i) {
                arr[j] = false;        
            }
        }
        for(int i=0; i<=n; i+=1) {
            if(true == arr[i]) {
                answer += 1;
            }
        }
        return answer;
    }
}

문제 출처: https://programmers.co.kr/learn/courses/30/lessons/12921?language=java

코딩테스트 연습 - 소수 찾기 | 프로그래머스