Swift - 백준 10773번 제로

제로 문제는 스택으로 구현하면 쉽게 풀 수 있습니다

 

 

0이 아니면 push 0이면 pop으로 구현할 수 있겠습니다.

let num = Int(readLine()!)!
var arr: [Int] = []

for _ in 1...num {
    let tmp = Int(readLine()!)!
    if tmp == 0 {
        arr.removeLast()
    } else {
        arr.append(tmp)
    }
}

var sum = 0

for i in arr {
    sum += i
}
print(sum)