227. Basic Calculator II

💡 풀이

// O(N)
var calculate = function (s) {
  let num = '';
  let sign = '';
  let stack = [];

  for (let i = 0; i <= s.length; i++) {
    let cur = s[i];
    // 숫자가 나올 때 (10의 자리를 넘어갈 때도 포함)
    if (!isNaN(s[i])) num += s[i];

    // 기호가 나올 때
    if (isNaN(s[i])) {
      num = Number(num);

      if (sign === '*') stack.push(stack.pop() * num);
      else if (sign === '/') stack.push(Math.trunc(stack.pop() / num));
      else if (sign === '+') stack.push(num);
      else if (sign === '-') stack.push(-num);
      else stack.push(num);
      sign = cur;
      num = '';
    }
    // console.log('num: ', num);
    // console.log('stack: ', stack);
  }
  //   console.log('answer: ', stack);
  //   console.log(stack.reduce((acc, cur) => acc + cur));
  return stack.reduce((acc, cur) => acc + cur);
};

let s = '14-3/2';
calculate(s);

📝 정리

어제에 이어서 stack을 이용한 문제였다. 해결하는데 매우 오랜 시간이 걸렸던 문제였다.(어제보다 더..!)

주의해야 할 부분이 곱셈, 나눗셈이 있을 때는 다른 기호보다 먼저 연산을 해줘야 했는데, 그렇기 때문에 곱하기 혹은 나누기가 나오면 stack 배열에 그 결과를 계산해서 바로 push하는 형태로 구현하였다.

수정, 지적을 환영합니다!

문제 링크

https://leetcode.com/problems/basic-calculator-ii/

LeetCode GitHub

https://github.com/tTab1204/LeetCode