🎲백준 2504번 괄호의 값

백준 2504번

💊 '(', '[' 괄호마다의 경우를 모두 쓰고, 올바르지 않은 경우도 하나하나 체크해준 코드는 시간초과 오류가 발생했다.
💊 다른 분의 코드를 참고하여 올바르지 않은 경우를 미리 확인하고, reverse, point 변수를 추가하여 코드를 간소화하였다.

시간 초과 코드

const fs = require('fs'); 
const input = fs.readFileSync('/dev/stdin').toString().trim().split('');

let stack = [];

for (let i = 0; i < input.length; i++) {
  if (input[i] === "]") {
    if (stack.length !== 0) {
      const top = stack[stack.length - 1];
      if (top === "[") {
        stack.pop();
        stack.push(3);
      } else if (top === "(") {
        console.log(0);
        return;
      } else {
        let sum = 0;
        while (true) {
          let top = stack.pop();
          if (top === "(") {
            console.log(0);
            return;
          } else if (top === "[") {
            stack.push(3 * sum);
            break;
          } else {
            sum += top;
          }
        }
      }
    } else {
      console.log(0);
      return;
    }
  } else if (input[i] === ")") {
    if (stack.length !== 0) {
      const top = stack[stack.length - 1];

      if (top === "(") {
        stack.pop();
        stack.push(2);
      } else if (top === "[") {
        console.log(0);
        return;
      } else {
        let sum = 0;
        while (true) {
          let top = stack.pop();
          if (top === "[") {
            console.log(0);
            return;
          } else if (top === "(") {
            stack.push(2 * sum);
            break;
          } else {
            sum += top;
          }
        }
      }
    } else {
      console.log(0);
      return;
    }
  } else {
    stack.push(input[i]);
  }
}

console.log(stack.reduce((a, b) => a + b, 0));

정답 코드

const fs = require('fs'); 
const input = fs.readFileSync('/dev/stdin').toString().trim().split('');

let stack = [];

for (let i = 0; i < input.length; i++) {
  let top = stack[stack.length - 1];
  if (top === "[" && input[i] === "]") {
    stack.pop();
  } else if (top === "(" && input[i] === ")") {
    stack.pop();
  } else {
    stack.push(input[i]);
  }
}

if (stack.length !== 0) {
  console.log(0);
  return;
} else {
  stack = [];
}

for (let i = 0; i < input.length; i++) {
  const top = stack[stack.length - 1];
  const reverse = input[i] === ")" ? "(" : "[";
  const point = input[i] === ")" ? 2 : 3;
  if (input[i] === "]" || input[i] === ")") {
    if (top === reverse) {
      stack.pop();
      stack.push(point);
    } else {
      let sum = 0;
      while (true) {
        let top = stack.pop();
        if (top === reverse) {
          stack.push(point * sum);
          break;
        } else {
          sum += top;
        }
      }
    }
  } else {
    stack.push(input[i]);
  }
}

console.log(stack.reduce((a, b) => a + b, 0));