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

문제 설명

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

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

 

제한 조건

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

입출력 예

 

입출력 예 설명

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

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

 

내 풀이

1.  n 이 2경우 answer는 무조건 1이므로 int answer = 1로 설정 

2. for문의 시작을 3부터 

3. 두번째 for문의 경우 범위를 j <= i 로 하는경우 시간 초과 발생 -> Math.sqrt를 사용해 제곱근까지 범위를 축소 

4. 소수의 경우 cnt = 0 이 되므로 cnt == 0 인 경우 answer++ 를 해준다.

class Solution {
    public int solution(int n) {
	    int answer = 1;

	for (int i = 3; i <= n; i++) {
		    int cnt = 0;
             for (int j = 2; j <= Math.sqrt(i); j++) {
				if (i % j == 0) {
					cnt++;
					break;
				}
			}
			if(cnt == 0) {
				answer++;
			}
		}
		return answer;
    }
}