[Python] Stack 구현하기

 Stack을 class로 구현해볼 것이다. 

파이썬의 list의 메소드를 가지고도 stack 처럼 쓸 수 있음

설계

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

10828번: 스택

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

위의 조건을 만족하도록 설계하였고 보통 isEmpty를 사용하여 list가 비었는지 확인함

class Stack:
    def __init__(self):
        self.stack_list = []

    def push(self, num):
        self.stack_list.append(num)

    def pop(self):
        if self.stack_list :
            return  self.stack_list.pop()
        else:
            return -1

    def size(self):
        return len(self.stack_list)

    def empty(self):
        if self.stack_list :
            return 0
        else:
            return 1

    def top(self):
        if self.stack_list:
            return self.stack_list[-1]
        else:
            return -1

    def view(self):
        return self.stack_list

---

stack1 = Stack()

print(stack1.pop())
print(stack1.top())
print(stack1.size())
stack1.push(5)
stack1.push(6)
stack1.push(8)
print(stack1.view())
print(stack1.pop())
stack1.push(123)
print(stack1.view())

-1
-1
0
[5, 6, 8]
8
[5, 6, 123]