[백준/Python] 10828. 스택

 

 

• push를 구현하기위해 append를 사용했고
• pop을 구현하기위해선 pop함수,
• size와 empty를 구현하기위해서는 len함수를 사용했다.
• 예외 사항들은 try, except 구문을 이용해 구현하였다

import sys
input = sys.stdin.readline # 시간단축을 위한 코드

N = int(input())
stack = []

for i in range(N):
    command = list(input().split())
    if command[0] == 'push':
        stack.append(command[1])
        
    elif command[0] == 'pop':
        try:
            print(stack.pop())
        except:
            print('-1')
            
    elif command[0] == 'size':
        print(len(stack))
        
    elif command[0] == 'empty':
        if len(stack):
            print('0')
        else:
            print('1')
            
    elif command[0] == 'top':
        try:
            print(stack[-1])
        except:
            print('-1')