알고리즘 스터디 5일차 - 문제 풀이(Javascript)

싱클베어·2022년 1월 19일

algorithm

목록 보기

5/6

아래 3개의 문제에 대해 문제 풀이를 작성하였습니다.

프로그래머스 - Summer/Winter Coding(~2018) - 소수 만들기
프로그래머스 - 2021 KAKAO BLIND RECRUITMENT - 신규 아이디 추천
프로그래머스 - Summer/Winter Coding(~2018) - 예산

1. 소수 만들기 (q37)

문제 설명

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

제한 조건

nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.

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

예시

입/출력

입력 [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을 만들 수 있습니다.

나의 풀이

function getcombination(array, selectNumber){
    // 조합은 서로 다른 n개의 원소를 가지는 집합에서 k개의 부분집합을 고르는 조합의 수.
    // 순서가 상관없다. C(n,k) = n! / (k!) * (n-k)!
    // nums = n개의 원소, 3개의 수 선택 k = 3
    // (nums.length) * (nums.length -1) * (nums.length-2) / (3 * 2 * 1) = 조합의 수
    
    // 조합을 만들때, 만드는 방식을 보면.. 4C3 일때
    // nums = [1,2,3,4] 일때, 가장 첫번째 값을 선택하고
    // 배열에 남은 [2,3,4] 중에 2개를 선택하여 조합 생성
    const results = [];
    
    if (selectNumber === 1) return array.map((value) => [value]); // 배열 마지막 선택 시 값 반환
    
    array.forEach((fixed, index, origin) => {
        const rest = origin.slice(index + 1); // index+1 값 이후 남은 배열 rest에 복제
        const combinations = getcombination(rest, selectNumber - 1); // 나머지에 대해서 조합을 구한다.
        const attached = combinations.map((combination) => [fixed, ...combination]); //  돌아온 조합에 떼 놓은(fixed) 값 붙이기
        results.push(...attached); // 배열 spread syntax 로 모두다 push
    });

  return results; // 결과 담긴 results return
}

function isPrime(num) {  // 소수 판별
    if(num===1){
        return false;
    } else if(num===2){
        return true;
    }
    
    if(num%2===0){
        return false;
    }
    
    let sqrt = parseInt(Math.sqrt(num)); // 제곱근 값을 구해 Int값으로 변경
    
    for (let i = 3; i<=sqrt; i++){
        if (num%i===0){
            return false;
        }
    }
    return true;
}

function solution(nums) {
    var answer = 0;
    let count = 0; // 소수 개수를 셀 count
    let result = 0;
    let combination_array = new Array();
    combination_array = getcombination(nums,3);
    console.log(combination_array);
    console.log(combination_array.length);
    for (let i=0; i<combination_array.length; i++){
        result = 0;
        for (let j=0; j<3; j++){
            result += combination_array[i][j];
        }
        isPrime(result) ? answer++ : false;    
    }
    
    
    
    // [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
    
    return answer;
}

조합을 찾는 문제인 것 까진 분석이 됐는데, 조합을 코드로 구현하는 부분에서 좀 헤맸다. 재귀 함수 내용이 나와 조합을 구하는 재귀 함수 샘플을 찾아 분석하고 공부하는데 좀 초점을 맞췄다. 효율을 위해 사람이 조합을 구하는 방식과 달리, 마지막에 선택할 수 있는 수가 단 한 가지인 경우를 먼저 if문으로 검사하는 구문이 함수 초반부에 나온다. 그 부분만 감안하고 코드를 읽으면 될 듯 하다.
그 외에는 소수를 구하는 함수를 별도로 지정하고, 조합으로 구한 이중 배열의 요소값을 모두 나온 값을 더해 소수인지 판별하고, 소수일 경우 answer의 카운트를 올려 그 값을 리턴한다.

2. 신규 아이디 추천 (q40)

문제 설명

카카오에 입사한 신입 개발자 네오는 "카카오계정개발팀"에 배치되어, 카카오 서비스에 가입하는 유저들의 아이디를 생성하는 업무를 담당하게 되었습니다. "네오"에게 주어진 첫 업무는 새로 가입하는 유저들이 카카오 아이디 규칙에 맞지 않는 아이디를 입력했을 때, 입력된 아이디와 유사하면서 규칙에 맞는 아이디를 추천해주는 프로그램을 개발하는 것입니다.
다음은 카카오 아이디의 규칙입니다.

아이디의 길이는 3자 이상 15자 이하여야 합니다.
아이디는 알파벳 소문자, 숫자, 빼기(-), 밑줄(_), 마침표(.) 문자만 사용할 수 있습니다.
단, 마침표(.)는 처음과 끝에 사용할 수 없으며 또한 연속으로 사용할 수 없습니다.
"네오"는 다음과 같이 7단계의 순차적인 처리 과정을 통해 신규 유저가 입력한 아이디가 카카오 아이디 규칙에 맞는 지 검사하고 규칙에 맞지 않은 경우 규칙에 맞는 새로운 아이디를 추천해 주려고 합니다.
신규 유저가 입력한 아이디가 new_id 라고 한다면,

1단계 newid의 모든 대문자를 대응되는 소문자로 치환합니다.
2단계 new_id에서 알파벳 소문자, 숫자, 빼기(-), 밑줄(), 마침표(.)를 제외한 모든 문자를 제거합니다.
3단계 new_id에서 마침표(.)가 2번 이상 연속된 부분을 하나의 마침표(.)로 치환합니다.
4단계 new_id에서 마침표(.)가 처음이나 끝에 위치한다면 제거합니다.
5단계 new_id가 빈 문자열이라면, new_id에 "a"를 대입합니다.
6단계 new_id의 길이가 16자 이상이면, new_id의 첫 15개의 문자를 제외한 나머지 문자들을 모두 제거합니다.
만약 제거 후 마침표(.)가 new_id의 끝에 위치한다면 끝에 위치한 마침표(.) 문자를 제거합니다.
7단계 new_id의 길이가 2자 이하라면, new_id의 마지막 문자를 new_id의 길이가 3이 될 때까지 반복해서 끝에 붙입니다.

예를 들어, new_id 값이 "...!@BaT#*..y.abcdefghijklm" 라면, 위 7단계를 거치고 나면 new_id는 아래와 같이 변경됩니다.

1단계 대문자 'B'와 'T'가 소문자 'b'와 't'로 바뀌었습니다.
"...!@BaT#*..y.abcdefghijklm" → "...!@bat#*..y.abcdefghijklm"

2단계 '!', '@', '#', '*' 문자가 제거되었습니다.
"...!@bat#*..y.abcdefghijklm" → "...bat..y.abcdefghijklm"

3단계 '...'와 '..' 가 '.'로 바뀌었습니다.
"...bat..y.abcdefghijklm" → ".bat.y.abcdefghijklm"

4단계 아이디의 처음에 위치한 '.'가 제거되었습니다.
".bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

5단계 아이디가 빈 문자열이 아니므로 변화가 없습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghijklm"

6단계 아이디의 길이가 16자 이상이므로, 처음 15자를 제외한 나머지 문자들이 제거되었습니다.
"bat.y.abcdefghijklm" → "bat.y.abcdefghi"

7단계 아이디의 길이가 2자 이하가 아니므로 변화가 없습니다.
"bat.y.abcdefghi" → "bat.y.abcdefghi"

따라서 신규 유저가 입력한 new_id가 "...!@BaT#*..y.abcdefghijklm"일 때, 네오의 프로그램이 추천하는 새로운 아이디는 "bat.y.abcdefghi" 입니다.

신규 유저가 입력한 아이디를 나타내는 new_id가 매개변수로 주어질 때, "네오"가 설계한 7단계의 처리 과정을 거친 후의 추천 아이디를 return 하도록 solution 함수를 완성해 주세요.

제한 조건

new_id는 길이 1 이상 1,000 이하인 문자열입니다.

new_id는 알파벳 대문자, 알파벳 소문자, 숫자, 특수문자로 구성되어 있습니다.

new_id에 나타날 수 있는 특수문자는 -_.~!@#$%^&*()=+[{]}:?,<>/ 로 한정됩니다.

예시

예시1)
"...!@BaT#*..y.abcdefghijklm" → "bat.y.abcdefghi"
문제의 예시와 동일.

예시2)
z-+.^." → "z--"

7단계를 거치는 동안 new_id가 변화하는 과정은 아래와 같습니다.

1단계 변화 없습니다.
2단계 "z-+.^." → "z-.."
3단계 "z-.." → "z-."
4단계 "z-." → "z-"
5단계 변화 없습니다.
6단계 변화 없습니다.
7단계 "z-" → "z--"

예시3)
"=.=" → "aaa"

7단계를 거치는 동안 new_id가 변화하는 과정은 아래와 같습니다.

1단계 변화 없습니다.
2단계 "=.=" → "."
3단계 변화 없습니다.
4단계 "." → "" (new_id가 빈 문자열이 되었습니다.)
5단계 "" → "a"
6단계 변화 없습니다.
7단계 "a" → "aaa"

예시4)
"123_.def" → "123_.def"

1단계에서 7단계까지 거치는 동안 newid(`"123.def"`)는 변하지 않습니다. 즉, new_id가 처음부터 카카오의 아이디 규칙에 맞습니다.

예시5)
"abcdefghijklmn.p" → "abcdefghijklmn"

1단계 변화 없습니다.
2단계 변화 없습니다.
3단계 변화 없습니다.
4단계 변화 없습니다.
5단계 변화 없습니다.
6단계 "abcdefghijklmn.p" → "abcdefghijklmn." → "abcdefghijklmn"
7단계 변화 없습니다.

나의 풀이

function solution(new_id) {    
    const answer = new_id
        .toLowerCase() // 대문자를 소문자로
        .replace(/[^\w-_.]/g, '') // -_.가 포함된 문자
        .replace(/\.{2,}/g, '.') // .이 2개 이상인 경우, '.' 로 변환
        .replace(/^\.|\.$/g, '') // 문장의 시작이 . 이거나 끝이 . 인 경우, ''로 변환
        .replace(/^$/, 'a') // 값이 없으면 new_id를 a로 입력
        .slice(0, 15).replace(/\.$/, ''); // new_id의 index 0부터 15자를 잘라오고, 끝자리에 .이 있는지 다시 확인
    const len = answer.length;
    return len > 2 ? answer : answer + answer.charAt(len - 1).repeat(3 - len);
}

정규식을 아주 진하게 배웠다. 언어마다 규칙이 조금씩 다르다고 하니 정규식 테스트를 할 수 있는 사이트 목록에서 테스트를 직접 해가며 대입했다.

정규식 테스트 사이트 모음

3. 예산 (q40)

문제 설명

S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.
물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.
부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.

제한 조건

d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.

d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.

budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.

예시

입/출력
[1,3,2,5,4] : budget = 9 , result = 3
[2,2,3,3] : budget = 10 , result = 4

입출력 예#1 :
각 부서에서 [1원, 3원, 2원, 5원, 4원]만큼의 금액을 신청했습니다. 만약에, 1원, 2원, 4원을 신청한 부서의 물품을 구매해주면 예산 9원에서 7원이 소비되어 2원이 남습니다. 항상 정확히 신청한 금액만큼 지원해 줘야 하므로 남은 2원으로 나머지 부서를 지원해 주지 않습니다. 위 방법 외에 3개 부서를 지원해 줄 방법들은 다음과 같습니다.

1원, 2원, 3원을 신청한 부서의 물품을 구매해주려면 6원이 필요합니다.

1원, 2원, 5원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.

1원, 3원, 4원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.

1원, 3원, 5원을 신청한 부서의 물품을 구매해주려면 9원이 필요합니다.

3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.

입출력 예#2 :
모든 부서의 물품을 구매해주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.

나의 풀이

function solution(d, budget) {
    var answer = 0;
    let count = 0; // 부서의 수
    let sum = 0; // budget 과 비교할, 부서예산의 합
    d.sort((a,b) => a-b); // 오름차순 정렬
    for (let i=0; i<d.length; i++){
        sum += d[i];
        count = d.length;
        if (sum > budget) {
            count = i; // 직전 부서까지는 부서 물품 구매가 가능하므로 i값 입력
            break;
        }
    }
    return count;
}

부서별로 신청한 금액 배열 d를 오름차순 정렬 해주고, 가장 낮은 부서별로 순차적으로 합계 sum 을 구해서 for문 마지막에서 예산 budget 보다 더 큰지를 꾸준히 체크해주면 되서 특별히 어려운 점은 없었다. 가장 많은 부서에게 사줘야한다는 전제 때문에 간단히 풀렸다. 물론, 실제 사례에서 이럴일은 없겠지만.