[Python]Stack 백준 10773

문제

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

앞에 문제에서 스택을 구현하였기에 필요한 부분만 가져왔다.

스택의 기본적인 사용을 알아보기 위한 문제인거 같다.

import sys
 
 
class Stack:
    def __init__(self):
        self.len = 0
        self.list = []
 
    def pop(self):
        pop_result=self.list[self.len-1]
        del self.list[self.len-1]
        self.len -= 1
        return pop_result
 
    def push(self,num):
        self.len += 1
        self.list.append(num)
 
 
n=int(input())
stack=Stack()
sum=0
while(n>0):
    n-=1
    command=sys.stdin.readline().split()
    if command[0]=='0':
        stack.pop()
    else:
        stack.push(command[0])
 
 
for i in stack.list:
    sum+=int(i)
 
 
print(sum)