백준 Swift [10773] 제로

난이도

실버 IV

 

문제

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

10773번: 제로

 

내 답안

var accountBook = [Int]()

for _ in 1...Int(readLine()!)! {
    let input = Int(readLine()!)!
    switch input {
    case 0:
        accountBook.removeLast()
    default:
        accountBook.append(input)
    }
}

print(accountBook.reduce(0, +))

• 빈 스택인 accountBook 초기화
• 횟수만큼 돌면서 입력값이 0인 것과 0이 아닌(default)것만 분기 처리를 하여 각각 removeLast, append를 수행
  (removeLast의 복잡도는 O(1))
• for문이 끝나면 reduce를 통해 모든 값을 더한 값을 출력하여 해결