Programmers / Level1 / 소수 만들기 / C++ / JS

문제 설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다.

숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때

소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

 

제한사항

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

---

입출력 예

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

 

입출력 예 설명

 

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

 

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

 

 

<C++> 2021.10.26

#include <vector>
#include <iostream>
using namespace std;

bool isPrime(int n) {
    if(n == 0 || n == 1) return false;
    
    for(int i=2; i*i<=n; i++) {
        if(n % i == 0)
            return false;
    }
    
    return true;
}

int solution(vector<int> nums) {
    int answer = 0;
    int len = nums.size();
    
    for(int i=0; i<len; i++) {
        for(int j=i+1; j<len; j++) {
            for(int k=j+1; k<len; k++) {
                int n = nums[i] + nums[j] + nums[k];
                
                if(isPrime(n) == true)
                    answer++;
            }
        }
    }
    
    return answer;
}

---

이 문제를 로직을 효율적으로 짜지 못한거같아서 수정이 필요할것 같다.

 

일단 본인이 짠대로 설명을 해보자면, 소수를 구하는 함수를 외부에 따로 두었고, 소수면 true를 반환하도록 bool형으로 만들었다.

 

그래서 3중 반복문을 이용해서 3개의 합이 소수일경우, answer++를 해주고 answer을 리턴한다.

---

 

<JS> 2022.05.01

function isPrime(n) {
    for(let i=2; i<=Math.sqrt(n); i++) {
        if(n % i == 0) return false;
    }
    return true;
}

function solution(nums) {
    let answer = 0;
    const len = nums.length;
    
    for(let i=0; i<len; i++) {
        for(let j=i+1; j<len; j++) {
            for(let k=j+1; k<len; k++) {
                let sum = 0;
                sum += nums[i] + nums[j] + nums[k]
                
                if(isPrime(sum)) answer++;
            }
        }
    }    
    
    return answer;
}