쥐수의 공부노트

백준 10773번 제로 본문

swift 알고리즘/스택

백준 10773번 제로

쥐수 2023. 6. 11. 17:18
728x90

정답 :

let n = Int(readLine()!)!
var pos = 0
var array : [Int] = []
var sum = 0

func push(num : Int) {
    array.append(num)
    pos += 1
}

func pop() {
    array.remove(at: pos - 1)
    pos -= 1
}
for _ in 0..<n {
    let num = Int(readLine()!)!
    
    if num == 0 {
        pop()
    } else {
        push(num: num)
    }
}

for i in array {
    sum += i
}
print(sum)
더보기

TMI : push 함수와 pop 함수를 구현할 수 있으면 쉽게 풀린다!

728x90

'swift 알고리즘 > 스택' 카테고리의 다른 글

백준 1874번 스택 수열  (0) 2023.06.11
백준 4949번 균형잡힌 세상  (0) 2023.06.11
백준 9012번 괄호  (0) 2023.06.11
백준 10828번 스택  (0) 2023.06.11