백준 10773 문제풀이 [python]

포스팅에 앞서 내용이 틀릴 수 있습니다.
해당 부분 지적 감사히 받습니다.

문제를 보자.

이번문제를 풀며 문제가 생겼다.

 

정답과 오답코드를 보자.

 

오답 코드

import sys as s

array = []

k = int(s.stdin.readline())

for i in range(k):

    num = int(s.stdin.readline())

    if num == 0 :

        array.remove(array[-1])

    else:

        array.append(num)

print(sum(array))

 

정답 코드

import sys as s

array = []

k = int(s.stdin.readline())

for i in range(k):

    num = int(s.stdin.readline())

    if num == 0 :

        array.pop()

    else:

        array.append(num)

print(sum(array))

 

필자가 확인해 본 결과 출력값은 같았다.

 

내 생각으론 그냥 스택, 큐 관련 문제에 속해있어서 관련 코드로 문제를 풀지 않으면 해결되지 않도록 한 것 같다.

 

관련코드로 풀자.