[프로그래머스 Lv.2] 소수 찾기 (js)

https://programmers.co.kr/learn/courses/30/lessons/42839?language=javascript 

코딩테스트 연습 - 소수 찾기

 

function solution(numbers) {

    const number = [...numbers];
    const answer = [];    
    
    for(let i = 1; i <= number.length; i++){
        permutation(number,i).forEach((v) => {
            isPrime(+v) ? answer.push(+v) : answer
        });
    }
    return [...new Set(answer)].length
}

function isPrime(num){
    let start = 2;
    while (start <= Math.sqrt(num)) {
        if (num % start++ < 1){
            return false;
        }
    }
    return num > 1;
}

function permutation(arr, selectNum){
        const result = [];
        if(selectNum === 1) return arr.filter((v)=> [v]);

        arr.forEach((v,idx,arr)=>{
            const fixer = v;
            const restArr = arr.filter((val, index)=> index !== idx);
            const permuArr = permutation(restArr, selectNum-1);
            const combineFixer = permuArr.map((v)=> fixer + v);
            result.push(...combineFixer);
        })
        return result;
    }

 

permutation 함수 => arr => 배열 selectNum => 개수 

 

                            selectNum만큼 조합한 결과를 리턴하는 함수 / 중복까지 전부 들어감 (011, 11, 0011) 나중에 처리

 

 

isPrime 함수 => 소수 체크 함수이다. 제곱근을 통해 시간복잡도를 줄일 수 있다.

 

 

solution에서 조합한 결과를 answer에 전부 담고 중복을 제거하기 위해 Set을 사용한 후 answer의 길이를 리턴하면 끝!