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

o 문제

 

https://school.programmers.co.kr/learn/courses/30/lessons/12921

프로그래머스

 

---

 

o 첫 번째 풀이 - 실패 ( 시간 초과 )

저번 약수 구하는 문제처럼 이 방식이 보편적으로 사용되지만, 코테에서 사용하면 시간초과가 나는 코드이다.

시간을 줄이는 알고리즘을 생각해봐야된다.

 

class Solution {
    public int solution(int n) {
        int result = 0;
        for (int i = 2; i <= n; i++) {
            int cnt = 0;
            for (int j = 1; j <= i; j++) {
                if (i % j == 0) {
                    cnt++;
                }
            }
            if (cnt <= 2) {
                result++;
            }
        }
        return result;
    }
}

 

 

생각해보니 약수 구하기와 비슷한 or 더 쉬운 알고리즘이다.

n의 루트값만 확인을 해보면 되는것이다.

 

o 최종 풀이

 

class Solution {
    public int solution(int n) {
        
        int answer = 0;
        for(int i = 2; i <= n; i++){
            boolean flag = true;
            for(int j = 2; j<= Math.sqrt(i); j++){
                if(i % j == 0){
                    flag = false;
                    break;
                
                }
            }
            if(flag){
                answer++;
            }
        }
        return answer;
    }
}