[알고리즘과 자료구조] JAVA - 소수 만들기

문제 설명

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

제한사항

nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.

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

문제풀이

- nums의 원소 안에 있는 값을 3개씩 더해주는 반복문을 만든다.

- decimal 소수 구하는 함수를 만들어서 소수일 경우 answer 리턴값을 ++ 하여 반환한다.

class Solution {
    public int solution(int[] nums) {
        int answer = 0;
        
        for ( int i = 0; i < nums.length; i++ ) {
            for ( int j = i+1; j < nums.length; j++ ) {
                for ( int k = j+1; k < nums.length; k++ ){
                    int num = nums[i] + nums[j] + nums[k];
                    if(decimal(num)){
                        answer++; 
                    }
                }
            }
        }
        
        return answer;
    }
    
    // 소수 구하는 함수
    static boolean decimal(int num){
        boolean result = true;
        for(int i = 2; i*i <= num; i++ ){
            if( num % i == 0) {
                result = false;
                break;
            }
        }
        return result;
    }
}