[BaeKJoon] 10828번: 스택 (Python)

1일 1문제 12일차!

오늘의 문제는 쉬어가는 타임! 스택입니다.

 

10828번: 스택

mystack = []
command_list = []

num = input()

for i in range(int(num)):
    command = input()
    command_list.append(command)
    
for command in command_list:
    cmd = command.split()
    
    if cmd[0] == 'push':
        mystack.append(cmd[1])
    elif cmd[0] == 'pop':
        if len(mystack) != 0:
            print(mystack.pop())
        else:
            print(-1)
    elif cmd[0] == 'top':
        if len(mystack) != 0:
            print(mystack[-1])
        else:
            print(-1)
    elif cmd[0] == 'size':
        print(len(mystack))
    elif cmd[0] == 'empty':
        if len(mystack) != 0:
            print(0)
        else:
            print(1)

SOMJANG/CODINGTEST_PRACTICE