[BOJ] (Swift) 1935 - 후위 표기식 2

1935번: 후위 표기식2

---

문제 요약

• 후위 표기식은 식의 앞에서부터 피연산자면 스택에 넣고 연산자면 스택에 들어있던 피연산자를 꺼내서 연산하는 것.
• 후위 표기식의 계산 결과를 소수점 둘째 자리까지 출력하라

 

풀이

• 피연산자가 A... Z로 주어지지만 결국에는 Double형이다.
• numbers에 입력 받은 숫자들이 들어가 있는데 A = numbers[0] , B = numbers[1] ... 이런 식으로 바꿔 주어야 한다.
• A...Z를 (아스키 코드 - 65)로 바꿔 index 번호로 바꿔주자.
• 연산 결과를 stack에 넣어 반복적으로 연산한다.

import Foundation

let N = Int(readLine()!)!
let input = readLine()!

var stack: [Double] = []
var numbers: [Double] = []

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

for char in input {
    if char.isUppercase {
        let index = Int(char.asciiValue! - 65)
        stack.append(numbers[index])
    } else {
        let first = stack.removeLast()
        let second = stack.removeLast()
        if char == "+" {
            stack.append(first + second)
        } else if char == "-" {
            stack.append(second - first)
        } else if char == "*" {
            stack.append(first * second)
        } else {
            stack.append(second / first)
        }
    }
}
print(String(format: "%.2f", stack[0]))