[C언어] 소수 만들기

문제 설명

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

제한사항

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

입출력 예

입출력 예 설명

• 입출력 예 #1
  [1,2,4]를 이용해서 7을 만들 수 있습니다.

• 입출력 예 #2
  [1,2,4]를 이용해서 7을 만들 수 있습니다.
  [1,4,6]을 이용해서 11을 만들 수 있습니다.
  [2,4,7]을 이용해서 13을 만들 수 있습니다.
  [4,6,7]을 이용해서 17을 만들 수 있습니다.

코드 구현

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// nums_len은 배열 nums의 길이입니다.
int solution(int nums[], size_t nums_len) {
    int answer = 0;
    int sum = 0;
    
    //배열에서 3개 골라서 더하는 반복문
    for(int i=0;i<nums_len;i++){
        for(int j=i+1;j<nums_len;j++){
            for(int k=j+1;k<nums_len;k++){
                sum = nums[i]+nums[j]+nums[k];
                int count = 0;
                //약수의 개수를 구하는 반복문
                for(int h=2;h<sum;h++){
                   if(sum%h==0)
                       count++;
                }
                if(count==0)
                    answer++;
            }
        }
    }
    return answer;
}

코드 설명

서로 다른 수 3개를 더하는 것이기 때문에 j와 k를 각각 i+1, j+1부터 시작하도록 지정하였다.
원소 3개를 더한 것을 sum에 저장하였고, sum의 크기만큼 다시 반복하면서 약수의 개수를 구하였다. 이때 h의 값을 2부터 sum-1까지 지정하였다. 1과 자기 자신만을 약수로 가지는 것이 소수이기 때문이다.
count값이 0이 아니라는 것은 1과 자기 자신을 제외하고 또 다른 약수가 존재한다는 뜻이므로 count가 0일 경우에만 answer의 값을 하나씩 증가시켰다.

추가

반복문이 사용되는 횟수를 줄일 수 있는 방법이 있을까싶어 다른 사람들의 코드를 찾아보았다.
하지만 다른 사람들도 3번 이상의 반복문을 사용하였고 다른 점이 있다면 약수의 개수를 판단하는 것을 또 다른 함수로 정의하였다는 것이다.

그리고 다른 코드를 찾아보다가 h의 범위를 너무 길게 sum-1까지 가지 말고 (sum/2)+1까지만 가도 충분하다는 것을 발견할 수 있었다.

즉, 코드로 표현하면 다음과 같다.

for(int h=2;h<(sum/2)+1;h++){
	if(sum%h==0)
    	count++;
}
if(count==0)
	answer++;