(Python) - BOJ(10828번) : 스택

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

10828번: 스택

---

• 해설 : 

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

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

 

 

 

---

• 풀이 :

각 case에 대해 스택의 기능을 사용하는 문제이다.

 

 

 

---

• 소스코드 : 

 

import sys
from collections import deque
input = sys.stdin.readline
 
if __name__ == "__main__":
    q = deque([])
    for _ in range(int(input())):
        q.append(input().split())
    stack = []
    while q:
        x = q.popleft()
        if x[0] == 'push':
            stack.append(x[1])
        elif x[0] == 'pop':
            if stack:
                print(stack.pop())
            else:
                print(-1)
        elif x[0] == 'size':
            print(len(stack))
        elif x[0] == 'empty':
            if stack:
                print(0)
            else:
                print(1)
        elif x[0] == 'top':
            if stack:
                print(stack[-1])
            else:
                print(-1)