[프로그래머스] 소수 만들기

✨️ 문제 설명

: 주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

제한사항

• nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
• nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.

🎲 자바 풀이

class Solution {
    public int solution(int[] nums) {
        int answer = 0;
        
        //3개의 숫자를 선택하고 더한후 -> 소수인지 확인해야함
        for(int idx = 0; idx < nums.length -2; idx++){
            for(int j = idx + 1; j < nums.length -1; j++){
                for(int k = j + 1; k < nums.length; k++){
                    int sum = nums[idx] + nums[j] + nums[k];
                    if(isPrime(sum)) answer++;
                }
            }
        }

        return answer;
    }
    
    //소수인지 판별하는 함수
    private boolean isPrime (int num){
        if(num <= 1) return false;
        for(int i = 2; i <= Math.sqrt(num); i++){
            if(num % i == 0) return false;
        }
        
        return true;
    }
}

풀이 설명

: 삼중 반복문으로 숫자를 세 개 선택하고, 더한 값을 소수인지 판별하였다.
: 일반적으로 소수를 판별할 때 2부터 N의 제곱근까지의 숫자로 나누면 빠르게 판별할 수 있기에, 제곱근을 구해 판별하였다. N이 제곱근보다 큰 약수를 가지고 있다면, 그보다 작은 약수도 반드시 존재하기 때문이다.