프로그래머스 -소수 찾기 LV1[java]

스브코·2021년 12월 3일
0
post-custom-banner

문제출처: https://programmers.co.kr/learn/courses/30/lessons/12921#

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

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

주어진 범위 안의 소수의 개수를 리턴하면 되는 문제이다.

입력 예시 : 10

출력 예시 : 4

1 ~ 10사이의 소수의 개수는 4개이다.

문제 풀이 1 [실패]

class Solution {
    public int solution(int n) {
        boolean [] notPrime = new boolean [1000001];
        int [] primeNums = {2, 3, 5, 7};
        int k = 0;
        while(k < primeNums.length) {
            for(int i = 2; i <= 100; i++) {
                if(i % primeNums[k] == 0) {
                    if(i != 2 && i != 3 && i != 5 && i != 7)
                        notPrime[i] = true;
                }
            }
            k++;
        }
        
        if(n <= 100)
            return counting(notPrime, n);
        
        primeCalc(notPrime, 100, 2);
        
        if(n <= 10000)
            return counting(notPrime, n);
        
        primeCalc(notPrime, 1000, 2);
        
        return counting(notPrime, n);
    }
    
    public static int counting(boolean[] notPrime, int n) {
        int count = 0;
        for(int i = 2; i <= n; i++) {
            if(!notPrime[i])
                count++;
        }
        return count;
    }
    
    public static void primeCalc(boolean[] notPrime, int num, int k) {
        int a = num + 1;
        if(num == 1000)
            a = num * 10 + 1;
         while(k < num) {
            if(!notPrime[k]) {
                for(int i = a; i <= num * num; i++) {
                    if(i % k == 0)
                        notPrime[i] = true;
                }
            }
            k++;
        }
        
    }
}

범위의 모든 소수를 다구했더니 시간초과가 나왔다.

문제 풀이 1 [성공]

class Solution {
    public int solution(int n) {
        int answer = 0;
        boolean [] notPrime = new boolean [n + 1];

        for(int i = 2; i < notPrime.length; i++) {
            if(!notPrime[i]) {
                int times = 2;
                for(int j = i * times;j < notPrime.length; j = i * times) {
                    notPrime[j] = true;
                    times++;
                }
            }
        }
        for(int i = 2; i < notPrime.length; i++) {
            if(!notPrime[i])
                answer++;
        }
        return answer;
    }
}

간단한 방법이 있었다...

profile
익히는 속도가 까먹는 속도를 추월하는 그날까지...
post-custom-banner

0개의 댓글