[ Python Coding ] 백준 10828. 스택

문제

입력 및 출력

실행 코드

import sys
input = sys.stdin.readline # 시간초과 문제 해결

n = int(input())

stack = []

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

풀이

• for문을 통해 입력된 문자에 해당하는 명령을 실행한다.
• input 대신 sys.stdin.readlin 을 작성해야 시간초과가 안난다.

[문제 링크]
https://www.acmicpc.net/problem/10828