Swift - 백준 4949번 균형잡힌 세상

이번 문제는 스택을 구현해서 쉽게 해결할 수 있습니다

 

import Foundation

var line = readLine()!
var arr: [Character]
while line != "." {
    arr = .init()
    for i in line {
        if i == "(" || i == "[" {
            arr.append(i)
        } else if i == ")" {
            if arr.last == "(" {
                arr.removeLast()
            } else {
                arr.append(i)
                break
            }
        } else if i == "]" {
            if arr.last == "[" {
                arr.removeLast()
            } else {
                arr.append(i)
                break
            }
        }
    }
    if arr.isEmpty {
        print("yes")
    } else {
        print("no")
    }
    line = readLine()!
}