[백준] (Swift) 1935번 - 후위 표기식2

 

문제링크

https://www.acmicpc.net/problem/1935

1935번: 후위 표기식2

 

후위 표기식을 계산하는 개념에 대해서 잘 알지 못하여 인터넷을 찾아보고, 후위표기식이 어떤 방식으로 계산되는지 먼저 살펴보았다. 기본적으로 stack 자료구조를 활용한다는 점이 특징이었다. 컴퓨터는 우리가 일반적으로 사용하는 중위표기식 (1+2=3) 보다 후위표기식이 더 이해가 쉬운지 알 수 있게 되었다.계속해서 결괏값을 stack에 저장해주기 때문이 아닐까.

내가 해당 문제를 풀이하기 위해 참고한 후위표기식 개념글은 아래와 같다.

https://todaycode.tistory.com/73?category=997273 

중위 표기법과 후위 표기법

 

위 글을 참고하여 백준 1935번의 예제 입력 1번을 풀어보면 아래와 같다!

 

내가 푼 풀이

• 입력받는 nums 에 무조건적으로 1,2,3,4 ..... 가 차례로 들어온다고만 생각하여 마지막에 아스키코드로 변경하여 64를 뺀 수를 stack에 넣어주었다. 당연히 오류였다. 사용자가 1.2.3.4.5 를 넣지 않고 2,45,677, .. 이렇게 무작위로 넣을 수도 있다는 사실을 깨닫고 입력받아놓은 nums의 인덱스에 접근하여 값을 stack에 넣는 방식으로 진행했다.
• 나머지 연산들은 구현하기 쉬웠다.
• 단 위에서 그림으로 그린 스택을 보면, - 와 / 는 순서가 중요하다. (여기서도 헤맸다.)
  • 2와 3을 곱할때 먼저꺼낸 3에다가 2를 곱하는게 아니라, 뒤늦게꺼낸 2에다가 3을 곱했다.
  • 뒷부분에 4와 5를 꺼내어 나눌때도, 5를 먼저꺼내어 5에다가 4를 나누는것이 아니라, 나중에꺼낸 4에다가 먼저꺼낸 5를 나눴다. 이처럼 순서가 중요하다.
  • 하지만 덧셈과 곱셈은 순서가 의미가 없기에, 나눗셈과 뺄셈에만 순서를 구별하기 위해 별도의 상수를 선언한 후, 계산을 실행했다.

import Foundation

let n = Int(readLine()!)!
var orderLine = readLine()!
var nums: [Double] = []
var stack: [Double] = []

for _ in 0..<n {
    let num = Double(readLine()!)!
    nums.append(num)
}

for order in orderLine {
    if order == "+" {
        stack.append(stack.removeLast()+stack.removeLast())
    } else if order == "-" {
        let a = stack.removeLast()
        stack.append(stack.removeLast() - a)
    } else if order == "*" {
        stack.append(stack.removeLast() * stack.removeLast())
    } else if order == "/" {
        let b = stack.removeLast()
        stack.append(stack.removeLast() / b)
    } else {
        let idx = order.asciiValue! - 65
        stack.append(nums[Int(idx)])
    }
}

print(String(format: "%.2f", stack[0]))