[JavaScript/section 6] 04 - 후위식 연산(postfix)

📌 04 - 후위식 연산(postfix)

후위연산식이 주어지면 연산한 결과를 출력하는 문제이다.

😂 나의 풀이 방법

처음엔 연산한 결과를 스택에 넣지 않고 answer 변수에 저장하는 방식으로 풀었다. 풀이 과정에서 answer = 0일 경우 연산이 이루어진 적이 없다는 것으로 생각하여 연산을 위해 pop() 함수를 두 번 이용해 배열에서 자료를 꺼낸다. 하지만 다음과 같이 '655-+'이 입력으로 주어진다면 첫 번째 연산의 결과가 0이 나오기 때문에 위 풀이 방법이 옳지 않다는 것을 알 수 있다. 연산한 결과를 스택에 넣는 방식으로 풀면 이러한 문제를 해결할 수 있다. 풀이 방법은 다음과 같다.

• for 문을 돌며 해당 값이 숫자인지 아닌지 검사한다. 이는 isNaN() 함수를 이용하면 된다.
• 숫자라면 스택에 해당 값을 숫자(Number)로 변환하여 넣어준다.
• 연산자라면 스택에서 자료를 꺼내 각각 rt(첫 번째로 꺼낸 자료), lt(두 번째로 꺼낸 자료) 변수에 저장한 후 조건에 따라 연산을 한 결과를 스택에 저장한다. 처음 스택에서 꺼낸 자료가 두 번째 피연산자에 해당한다❗
• for 문이 끝나면 스택의 첫 번째 인덱스에 해당하는 값이 연산한 결과이다.

 

📝 풀이

function solution(s) {
  let answer;
  let stack = [];

  for (let x of s) {
    if (!isNaN(x)) stack.push(Number(x));
    else {
      let rt = stack.pop();
      let lt = stack.pop();

      if (x === '+') stack.push(lt + rt);
      else if (x === '-') stack.push(lt - rt);
      else if (x === '*') stack.push(lt * rt);
      else if (x === '/') stack.push(lt / rt);
    }

    answer = stack[0];
  }
  return answer;
}

let str = '655-+';
console.log(solution(str));