LeetCode 394 - Decode String (Medium)

문제

LeetCode - 394번

풀이 과정

문자열의 규칙에 따라 디코딩하여 원본 문자열을 찾는 문제입니다.
[ 괄호 앞에는 반복할 숫자가 주어지기 때문에 스택 을 활용하여 문제를 해결하였습니다.

먼저 문자열의 각 문자를 순회하며 [ 와 ] 그리고 숫자 를 제외한 문자들은 그냥 push 해줍니다.
만약 숫자를 만날 경우에는 현재 스택의 top 이 숫자인지 판단하고 숫자라면 해당 숫자에 이어붙입니다.
(두 자리수 이상이 있을 수 있기 때문에)

이후 ] 를 만나면 [ 가 발견될 때 까지 문자들을 pop 하고 [ 앞의 숫자만큼 repeat 해줍니다.
최종적으로 스택 에 남아있는 문자들을 모두 합쳐주면 우리가 찾는 원본 문자열이 됩니다.

코드

/**
 * @param {string} s
 * @return {string}
 */
var decodeString = function (s) {
  const stack = [];

  for (let c of s) {
    if (c === "]") {
      let temp = "";
      while (stack[stack.length - 1] !== "[") {
        temp = stack.pop() + temp;
      }
      stack.pop();
      temp = temp.repeat(parseInt(stack.pop()));
      stack.push(temp);
    } else if (!isNaN(c)) {
      if (!isNaN(stack[stack.length - 1])) {
        stack[stack.length - 1] += c;
      } else {
        stack.push(c);
      }
    } else {
      stack.push(c);
    }
  }

  return stack.join("");
};