BOJ(백준) 10828번 스택 파이썬

10828번: 스택

---

코드

stack = []
def push(X):
    stack.append(X)

def pop():
    if len(stack) != 0:
        return stack.pop()
    else:
        return -1

def size():
    return (len(stack))

def empty():
    if stack:
        return 0
    else:
        return 1

def top():
    if stack:
        return stack[-1]
    else:
        return -1

import sys
r = sys.stdin.readline
N = int(r())
for _ in range(N):
    f = r().rstrip()
    if f[:2] == 'pu':
        push(f[5:])
    elif f[:2] == 'po':
        print(pop())
    elif f[:2] == 'si':
        print(size())
    elif f[:2] == 'em':
        print(empty())
    else:
        print(top())

 

해설

LIFO(Last In First Out)의 특징을 갖는 스택에 대한 기본적인 문제이다. 파이썬에서 리스트의 마지막 요소를 반환하면서 동시에 삭제하는 pop() 의 용도를 알고 있으면 쉽게 풀 수 있는 문제라고 생각한다. 주어진 다섯 가지 기능들(push, pop, size, empty, top)을 함수로 구현해서 코드를 작성하였다.

 

결과