프로그래머스 - 둘만의 암호

✨ Lv. 1 - 둘만의 암호

문제 링크 : https://school.programmers.co.kr/learn/courses/30/lessons/155652

문제 설명

두 문자열 s와 skip, 그리고 자연수 index가 주어질 때, 다음 규칙에 따라 문자열을 만들려 합니다. 암호의 규칙은 다음과 같습니다.

• 문자열 s의 각 알파벳을 index만큼 뒤의 알파벳으로 바꿔줍니다.
• index만큼의 뒤의 알파벳이 z를 넘어갈 경우 다시 a로 돌아갑니다.
• skip에 있는 알파벳은 제외하고 건너뜁니다.

예를 들어 s = "aukks", skip = "wbqd", index = 5일 때, a에서 5만큼 뒤에 있는 알파벳은 f지만 [b, c, d, e, f]에서 'b'와 'd'는 skip에 포함되므로 세지 않습니다. 따라서 'b', 'd'를 제외하고 'a'에서 5만큼 뒤에 있는 알파벳은 [c, e, f, g, h] 순서에 의해 'h'가 됩니다. 나머지 "ukks" 또한 위 규칙대로 바꾸면 "appy"가 되며 결과는 "happy"가 됩니다.

두 문자열 s와 skip, 그리고 자연수 index가 매개변수로 주어질 때 위 규칙대로 s를 변환한 결과를 return하도록 solution 함수를 완성해주세요.

제한사항

• 5 ≤ s의 길이 ≤ 50
• 1 ≤ skip의 길이 ≤ 10
• s와 skip은 알파벳 소문자로만 이루어져 있습니다.
• skip에 포함되는 알파벳은 s에 포함되지 않습니다.
• 1 ≤ index ≤ 20

풀이 코드 + 설명

우선 각 알파벳이 skip에 포함되지 않은 알파벳인지를 판단하기 위해 배열을 생성하여 이를 저장하였습니다. 이후, s를 순회하며 각 알파벳마다 다음을 반복하였습니다.

• 현재 알파벳을 ASCII 코드로 변경
• 반복문을 이용하여 index 만큼 1씩 증가시켜 알파벳을 변경
• 단, skip할 알파벳인 경우에는 for문의 i를 감소

이에 대한 코드는 아래와 같습니다.

function solution(s, skip, index) {
    let alpha = new Array(26).fill(true);
    [...skip].forEach((v) => alpha[v.charCodeAt() - 97] = false);
    
    return [...s].map((v) => {
        let current = v.charCodeAt() - 97;
        for(let i = 0; i < index; i++) {
            current = (current + 1) % 26;
            if(!alpha[current]) i--;
        }
        return String.fromCharCode(current % 26 + 97);
    }).join('');
}

알파벳 소문자로만 이루어진 s이기 때문에 더 간단한 방법으로 해결할 수 있습니다. skip을 제외한 오름차순으로 정렬된 알파벳 배열을 생성하여, 이를 바탕으로 index만큼 더한 위치의 알파벳으로 교체하여 return 합니다.

function solution(s, skip, index) {
    const alpha = [...'abcdefghijklmnopqrstuvwxyz'].filter((v) => !skip.includes(v));
    return [...s].map((v) => alpha[(alpha.indexOf(v) + index) % alpha.length]).join('');
}