0411 소수 만들기

문제 설명

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

제한사항

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

입출력 예

코드

class Solution {
	public int solution(int[] nums) {
    	int answer = -1;
    	int a=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++){
                	a=nums[i]+nums[j]+nums[k];
                	answer++;
                
                	if(a%k==0){
                   		answer--;
                	}    
            	}
        	}
    	}
    	return answer;
	}
}

이 코드는 [1,2,3,4],[1,2,7,6,4]보다 큰 수가 들어오면 에러가난다.

다시 푼 코드

class Solution {
	public int solution(int[] nums) {
    	int answer = 0;
    	int a=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++){
                	a=nums[i]+nums[j]+nums[k];
                
                	int cnt=0;
                
                    for(int b=1; b<=a; b++){
                    	if(a%b==0){
                        	cnt++;
                    	}
                	}
                	if(cnt==2){
                    	answer+=1;
                	}
            	}
        	}
    	}
    	return answer;
	}
}

풀이

a에 3개의 숫자를 더한 다음에 for문을 돌려 a%b가 0일때 cnt++를 해준다. 그리고 cnt가 2일때 answer에 1을 더해준다