[day5] 백준 - 스택 제로 (python)

좀 쉬운 것 같아서 하나 더 풀기로 했다. 근데 이것도 거의 10분 컷이었다. 완전 쉬움

 

아까 살짝 맛본 class를 사용해보았다.

import sys

class stack:
  def __init__(self):
    self.L = []
    
  def push(self, X):
    self.L.append(X)

  def pop(self):
    if len(self.L) == 0:
      return -1
    else:
      self.L.pop()
    
  def sum(self):
    return sum(self.L)

a = stack()
N = int(sys.stdin.readline())

for i in range(N):
  L = list(map(str, sys.stdin.readline().strip().split()))
  if L[0] == "0": # 처음에 0 과 비교해서 값이 틀렸다. 입력값은 문자열로 받기 때문에 "0"과 비교해야함
    a.pop()
  else: 
    a.push(int(L[0]))

print(a.sum())