[프로그래머스 / Level1] 소수 만들기 (Java)

문제

https://programmers.co.kr/learn/courses/30/lessons/12977?language=java 

코딩테스트 연습 - 소수 만들기

 

풀이

• nums 배열에서 서로 다른 3개의 수를 더한 결과(sum)가 소수인 경우를 카운트
• 서로 다른 3개의 수를 차례로 선택하는 과정을 각각 for문으로 구현. 
  1. 첫번째 수 a를 선택하는 for문(nums[i])
  2. 두번째 수 b를 선택하는 for문(nums[j])
  3. 세번째 수 c를 선택하는 for문(nums[k])
• 소수 판별 조건은 "2부터 자기자신의 제곱근까지 수 중 나누어 떨어지는 값이 있는가?" => true/false
  Math.sqrt(int n) : n의 제곱근을 반환

 

기존 코드 (성공)

class Solution {
    public int solution(int[] nums) {
        int answer = 0;
        
        for(int i = 0; i < nums.length - 2; i++){
            for(int j = i + 1; j < nums.length - 1; j++){
                for(int k = j + 1; k < nums.length; k++){
                    int sum = nums[i] + nums[j] + nums[k];
                    if(isPrime(sum)){
                        answer++;
                    }
                }
            }
        }
        
        return answer;
    }
    
    public boolean isPrime(int num){
        for(int i = 2; i <= Math.sqrt(num); i++){
            if(num % i == 0){
                return false;
            }
        }
            
        return true;
    }
}

 

• 소수판별, 배열에서 특정 요소 카운트 등의 단독 기능들은 메서드화 해서 빼주는 것이 좋다.
  ex) isPrime(int num) / countPrimes(int[] arr) 등