백준 10828 파이썬 (스택)

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

💻 Code

import sys
n = int(sys.stdin.readline())

stack=[]
for i in range(n):
    
    command = sys.stdin.readline().split()  
    if command[0] == "push":
        stack.append(command[1])
    elif command[0] == "pop":
        if(len(stack)==0):
            print(-1)
        else:
            print(stack.pop())
    elif command[0] == "size":
        print(len(stack))
    elif command[0] == "empty":
        if(len(stack)==0):
            print(1)
        else: 
            print(0)
    elif command[0] == "top":
        if(len(stack)==0):
            print(-1)
        else:
            print(stack[-1])
            

• 파이썬에서 스택은 리스트를 활용하자!

• command = sys.stdin.readline().split()에서 command는 따로 명시 안 해줘도 자동으로 리스트 자료형이 된다.