[프로그래머스 JS] 둘만의 암호

📖 문제 설명

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

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

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

 

---

🚫 제한 사항

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

---

💾 입출력 예시

 

s	skip	index	result
"aukks"	"wbqd"	5	"happy"

 

---

⌨️ 나의 풀이 (코드)

 

코드를 작성하기에 앞서 의사 코드를 통해 생각을 정리해 보았다.

 

1. s를 기반으로 index만큼 전진 
2. 전진할 때 skip의 문자열이 있다면 +1 만큼 더 전진
3. 순회하며 각 문자의 값을 모아 종료 시 반환

 

생각보다 간단한 과정 같아 보였다. 바로 코드로 작성해 보았는데...

 

function solution(s, skip, index) {
  const alphabet =["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z"];
  let answer = "";

  for (char of s) {
    let letterIdx = alphabet.indexOf(char);
    let idx = index;
    
    for (let i = 1; i < index + 1; i++) {
      if (skip.includes(alphabet[letterIdx + i])) {
        idx++;
      }
    }
    answer += alphabet[(letterIdx + idx) % 26];
  }
  return answer;
}

 

기본 테스트는 통과했지만 추가적인 테스트 케이스들이 모두 실패했다.

 

잠시 고민해보니 생각했던 코드의 로직에는 문제가 있었다.

 

먼저 i 만큼 반복하지만, 배열 alphabet의  letterIdx + i 번째 인덱스 값이 skip에 해당할 경우 본문의 내용처럼 해당 문자를 생략하고 다음 글자를 한 번 더 확인하는 과정이 이루어지고 있지 않았다.

 

따라서 letterIdx + i 에서 letterIdx + idx로 변경 하고 idx는 index 값에서 증가하는 값에서 순회와 별개로 alphabet 배열 내 요소의 탐색 거리를 결정하는 변수로 바꿔주었다. 추가로 변경 사항이 제대로 반영되기 위해 조건문에 i를 1 감소시키고, idx의 값은 그대로 1을 증가하는 방식으로 변경했다.

 

마지막으로 answer에 추가할 문자의 경우 현재 탐색중인 다음 글자가 아닌 해당 글자이므로 letterIdx + idx에서 1을 빼주었다.

 

function solution(s, skip, index) {
  const alphabet =["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z"];
  let answer = "";

  for (char of s) {
    let letterIdx = alphabet.indexOf(char);
    let idx = 1;
    for (let i = 1; i < index + 1; i++) {
      const nextChar = alphabet[(letterIdx + idx) % 26];
      if (skip.includes(nextChar)) {
        i--;
        idx ++;
      } else {
        idx++;
      }
    }
    answer += alphabet[(letterIdx + idx - 1) % 26];
    idx = 0;
  }
  return answer;
}

 

로직 수정에 있어 기존 변수의 이름을 그대로 채용해 약간 헷갈렸지만, 침착하게 전체 코드를 훑어보고 문제를 해결할 수 있었다.

 

---

📝 Note

 

훨씬 간결하고 합리적인 코드가 있었다.

 

function solution(s, skip, index) {
    const alphabet = ["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", 
                      "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", 
                      "u", "v", "w", "x", "y", "z"].filter(c => !skip.includes(c));
    return s.split("").map(c => alphabet[(alphabet.indexOf(c) + index) % alphabet.length]).join("");
}

 

먼저 alphabet 배열에서 .filter 기능을 사용해 skip에 포함된 문자를 배열에서 걸러낸다.

 

이후 탐색이 시작되는 문자열 s를 배열로 바꾸고, 요소 하나하나, 즉 각 문자를 index만큼 증가시킨 요소에서 찾아내 합치는 방법이다.

 

alpabet 배열의 길이가 제한되어 있어 합리적인 코드라고 생각했다.