[BOJ 10828] 스택 - 파이썬 풀이

 

문제 보기

10828번: 스택

문제 해설

정수를 저장하는 스택을 구현하고 push, pop, size, empty, top 5가지 기능을 구현하는 문제입니다.

스택에 대한 설명은 아래 링크 클릭!

파이썬(Python)으로 스택과 큐 사용하는 법 / 예제 포함

 

파이썬(Python3) 코드

import sys
import collections

stack = collections.deque()

n = int(sys.stdin.readline().strip())
for _ in range(n):
    command = sys.stdin.readline().strip()
    if command[:2] == "pu":
        num = int(command.split(' ')[1])
        stack.append(num)
    if command[:2] == "po":
        if not stack:
            print(-1)
            continue
        print(stack.pop())
    if command[0] == 's':
        print(len(stack))
    if command[0] == 'e':
        if stack:
            print(0)
            continue
        print(1)
    if command[0] == 't':
        if not stack:
            print(-1)
            continue
        print(stack[-1])

직접 스택을 구현해보는 연습을 하는 것도 좋지만, 알고리즘 문제를 풀 때는 deque를 사용하는 것이 효율적입니다.

또한 사용자로부터 입력을 받을 때, input()보다는 sys.stdin.readline().strip()을 사용하면 더 빨라집니다.

 

[BOJ] 파이썬(Python) 주의사항 및 Tips