[Python : 백준 10828] 스택

문제

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

풀이

단순 구현 문제.
명령어별 예외처리에 주의해서 구현하면 된다.
#1, #2 로 구분해서 구현해봤는데 모두 잘 되는 것 같다.

lines = int(input())
commands = []
stack = []
for i in range(lines):
    commands.append(input())

for command in commands:
    if command[0:3] == "pus":
        stack.append(int(command.split()[1]))
    
    #1 if else로 print구분
    if command[0:3] == "pop":           
        if stack: print(stack.pop())
        else: print(-1)
    #2 print함수 내에서 if else 사용
    if command[0:3] == "siz": print(len(stack))
    if command[0:3] == "top": print(stack[-1] if stack else -1)
    if command[0:3] == "emp": print(0 if stack else 1)