[백준] 10828번 파이썬 (스택)

1. 문제와 예제

문제

예제

---

2. 전체 코드

import sys
input = sys.stdin.readline

# 입력 받기
N = int(input().rstrip()) # 명령의 수
stack = [] # 스택

for _ in range(N):
    com = input().rstrip() # 명령
    # push X
    if (com.split()[0] == 'push'):
        stack.append(com.split()[1])
    # pop
    elif (com == 'pop'):
        l = len(stack)
        if (l == 0):
            print("-1")
        else:
            print(stack.pop(l-1))
    # size
    elif (com == 'size'):
        print(len(stack))
    # empty
    elif (com == 'empty'):
        if (len(stack) == 0):
            print(1)
        else:
            print(0)
    # top
    elif (com == 'top'):
        l = len(stack)
        if (l == 0):
            print("-1")
        else:
            print(stack[l-1])

---

3. 풀이

stack의 구조를 활용한 가장 기초적인 문제다.