10828 스택

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

10828번: 스택

 

내 풀이(맞음)

class Stack:
    def __init__(self, ary):    
        self.ary = []
    def push(self, ary, item):
        ary.append(item)
    def pop(self, ary):
        if ary:
            t = ary[len(ary)-1]
            del ary[len(ary)-1]
            return t
        else:
            return -1
    def size(self, ary):
        return len(ary)
    def empty(self, ary):
        if ary:
            return 0
        else:
            return 1
    def top(self, ary):
        if ary:
            return ary[len(ary)-1]
        else:
            return -1
a = []
rst = []
s = Stack(a)
n = int(input())
for i in range(n):
    x = input().split()
    if x[0] == 'push':
        s.push(a, x[1])
    elif x[0] == 'pop':
        rst.append(s.pop(a))
    elif x[0] == 'size':
        rst.append(s.size(a))
    elif x[0] == 'empty':
        rst.append(s.empty(a))
    elif x[0] == 'top':
        rst.append(s.top(a))
for i in rst:
    print(i)

Stack 클래스를 만들고 __init__로 초기화 함수를 선언했습니다. push는 ary에 추가하는 함수이고 pop은 ary에 내용이 있다면 t에 맨 위 값을 저장하고 삭제해준 다음 t를 리턴합니다. size는 스택의 개수를 구하고 empty는 스택에 값이 존재하면 0 아니면 1을 리턴합니다. 마지막으로 top는 가장 위에 값을 리턴합니다.