728x90
문제 설명
주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다.
숫자들이 들어있는 배열 nums가 매개변수로 주어질 때,
nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는
경우의 개수를 return 하도록 solution 함수를 완성해주세요.
제한 사항
- nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
- nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.
문제 풀이 방법
- 이 문제는 우선 소수 판별 함수를 한개 미리 선언하고 그 다음에 문제를 풀었다.
- DFS를 적용해 문제를 풀었다.
- 소수임이 판별되면 answer에 1을 더하도록 해서 문제를 해결했다.
내가 작성한 코드
const isPrime = (num) =>{
for(let i=2; i<=Math.sqrt(num); i++){
if(num % i === 0){
return false;
}
}
return true;
}
const solution = (nums) => {
let answer = 0;
for(let i=0; i<nums.length; i++){
for(let j=i+1; j<nums.length; j++){
for(let k=j+1; k<nums.length; k++){
const num = nums[i] + nums[j] + nums[k];
if(isPrime(num)){
answer += 1;
}
}
}
}
return answer
}728x90
