백준 10773번- '제로' (PYTHON 풀이)

문제:

 

어제에 이어 스택 카테고리 문제를 또 풀었다. 오늘 문제는 어제보다도 더 간단하다.

예시와 같이 입력으로 0이 들어올 때는 기존 스택에 쌓아놨던 수 중 마지막 수를 삭제하고 0 이외의 수를 입력받았을 땐 그 수를 스택에 그대로 쌓는다. 그 후 출력은 스택에 존재하는 값을 전부 더한 후 이를 출력하면 된다.

 

풀이:

K = int(input())
stack = []

for i in range(K):
    a = int(input())
    if a == 0:
        tra = stack.pop()
    else:
        stack.append(a)

print(sum(stack))

답은 다음과 같다.

pop() 함수를 활용하면 정말 간단하게 풀 수 있다. 입력값이 0이면 pop() 함수를 활용해서 마지막 수를 빼버리고, 그 외의 수를 append() 함수를 통해 리스트에 그대로 넣는다. 그 후 sum()을 활용해 리스트에 존재하는 모든 수를 더해서 출력하면 끝이다!