문제출처: https://programmers.co.kr/learn/courses/30/lessons/12921#
1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요.
소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다.
(1은 소수가 아닙니다.)
주어진 범위 안의 소수의 개수를 리턴하면 되는 문제이다.
입력 예시 : 10
출력 예시 : 4
1 ~ 10사이의 소수의 개수는 4개이다.
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++;
}
}
}
범위의 모든 소수를 다구했더니 시간초과가 나왔다.
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;
}
}
간단한 방법이 있었다...