백준[10828] python

문제 :

아주 간단한 문제로 왼쪽에 명령어가 나오고, 오른쪽엔 어떤 값이 주어진다.(사실 push 할 값뿐임)

필자는 파이썬을 class를 이용해서 아주 간단하게 구현했다. 

import sys

class stack:
    def __init__(self):
        self.data = []
    def push(self,x):
        self.data.append(x)
    def pop(self):
        if self.data:
            return self.data.pop()
        else:
            return -1
    def size(self):
        return len(self.data)
    def empty(self):
        return 1 if self.size() == 0 else 0
    def top(self):
        if self.data:
            return self.data[-1]
        else:
            return -1
n = int(sys.stdin.readline())
s = stack()

for _ in range(n):
    input_ = sys.stdin.readline().rstrip().split()
    o = input_[0]
    if o == 'push':
        s.push(input_[1])
    elif o =='top':
        print(s.top())
    elif o =='empty':
        print(s.empty())
    elif o =='size':
        print(s.size())
    elif o =='pop':
        print(s.pop())