문제 출처: https://programmers.co.kr/learn/courses/30/lessons/12977
주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.
입력
[1,2,7,6,4]
출력
4
입출력 예
[1,2,4]를 이용해서 7을 만들 수 있습니다.
[1,4,6]을 이용해서 11을 만들 수 있습니다.
[2,4,7]을 이용해서 13을 만들 수 있습니다.
[4,6,7]을 이용해서 17을 만들 수 있습니다.
import java.util.*;
class Solution {
public int solution(int[] nums) {
int answer = 0;
for(int i = 0; i < nums.length; i++) {
int n1 = nums[i];
for(int j = i + 1; j < nums.length; j++) {
int n2 = n1 + nums[j];
for(int k = j + 1; k < nums.length; k++) {
int n3 = n2 + nums[k];
if(checkPrime(n3))
answer++;
}
}
}
return answer;
}
public static boolean checkPrime(int n) {
for(int i = 2; i <= n / 2; i++) {
if(n % i == 0)
return false;
}
return true;
}
}
다른 풀이를 보니 prime Number를 확인 할때
for(int i = 1; i <= (int)Math.sqrt(num); i ++){
범위를 더 좁혔으면 좋았을것 같다.