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

프로그래머스

 

CODE

import java.util.*;
class Solution {
    public int solution(int[] nums) {
        int answer = 0;
        
        Arrays.sort(nums);

        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 result = isPrime(nums[i], nums[j], nums[k]);
                    answer += result;
                }
            }
        }
        return answer;
    }
    
        public static int isPrime(int x, int y, int z) {
        int sum = x + y + z;
        
        int cnt = 0;
        for(int i = 3; i < sum; i+=2) {
            if(sum % i == 0 || sum % 2 == 0) {
                return 0;
            }
        }
        return 1;
    }
}

소수를 판별하는 문제로, 코딩테스트에서 자주 출제되는 유형 중 하나이다. 소수는 1부터 자기 자신의 숫자 범위에서 1과 자기 자신만을 약수로 가지는 수이다. for문은 3중 for문을 이용했으며, 소수인지 아닌지 판단하는 로직은 따로 함수로 뺐다.

소수 문제에서 나름 핵심이라고 생각하는 부분은 소수를 판별하는 for문의 반복 횟수를 어떻게 줄이느냐에 있다고 생각한다(물론 위 코드가 엄청난 효율성을 자랑하는 코드는 아니고 나름대로 예외처리 한 것). 나는 i를 0이나 1이 아닌 3부터 시작하였고 i를 2씩 증가시켰다. 반복문을 돌 때 한 번이라도 나누어 떨어지거나, 혹은 짝수일 때 바로 반복문을 종료하도록 로직을 작성했다.

 

실행 결과