[BOJ / 백준] 10828번 스택 python / 파이썬

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

10828번: 스택

---

입력을 문자열 리스트로 받아준다.

그래서 그 리스트의 -1번째 인덱스 ( 1번째 인덱스랑도 같음)의 단어랑  push, pop, size, empty, top를 비교해주고

해당 단어에 해당되는 동작을 수행시켜준다.

 

import sys

N = int(sys.stdin.readline())
stack = []
for i in range(N):
    word = list(map(str, sys.stdin.readline().split()))
    if len(word) == 2:
        word[-1] = int(word[-1])
    if word[0] == 'push':
        stack.append(word[-1])
    elif word[0] == 'pop':
        if stack == []:
            print(-1)
        else:
            print(stack.pop())
    elif word[0] == 'size':
        print(len(stack))
    elif word[0] == 'empty':
        if stack == []:
            print(1)
        else:
            print(0)
    elif word[0] == 'top':
        if stack == []:
            print(-1)
        else:
            print(stack[-1])

import sys 해주고 sys.stdin.readline으로 입력을 받지 않으면 시간 초과 발생!!