[Swift] 백준 - 제로 ( 스택 )

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

 

 

//
//  main.swift
//  bjbj
//
//  Created by 양희태 on 9/9/24.
//
import Foundation

func main() {
   
    let N = Int(readLine()!)!
    
    var stack : [Int] = []
    
    for _ in 0..<N {
        let num = Int(readLine()!)!
        
        switch num{
        case 0:
            stack.removeLast()
        default:
            stack.append(num)
            
        }
    }
    
    let total = stack.reduce(0, +)
    print(total)
 
    
}

// main 함수 실행
main()