[백준][Python] 10828번 스택

10828번: 스택

 

코드

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

 

설명

 

파이썬에서 스택은 리스트를 이용하여 구현 가능하다.

 

for문 안에서 입력 값을 받을 때 input().split()를 사용하면 시간초과 오류가 뜬다.

 

리스트 가장 오른쪽에 있는 인덱스 값은 -1로도 표현 가능하다.