프로그래머스 LV1. 소수 만들기 (자바)

🍀 문제

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

 

입출력 예

[1,2,3,4]	1
[1,2,7,6,4]	4

 

 

 

코딩테스트 연습 - 소수 만들기 | 프로그래머스 (programmers.co.kr)

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

 

 

😊 나의 코드

class Solution {
    public int solution(int[] nums) {
        return combination(nums,0,0,3);
    }
	
	public int combination(int[] arr,int sum,int depth, int r) {
		if(r==0) {// 조합이 끝난 경우
			// 소수인지 확인
			for(int i=2; i<sum; i++) {
				if(sum%i==0) {
					return 0; //소수가 아님
				}
			}
			return 1;// 소수인 경우 결과에 1 더하기
		}else if(depth == arr.length) {//다 확인했지만 조합이 안된 경우
			return 0;
		}else {
			int result = combination(arr,sum,depth+1,r);//선택 안한 경우
			result += combination(arr,sum+arr[depth],depth+1,r-1);//선택한 경우
			return result;
		}	
	}
}

 

 

 

⭐ 조합을 사용해서 풀었다.

조합을 하려니 재귀를 이용해야했고 lv1 치고는 어려운 것 같다ㅠㅠ

조합 참고한 블로그 👇

https://minhamina.tistory.com/38