백준 / 28278번 / 스택 2 / python 파이썬

 

문제 : 28278번: 스택 2 (acmicpc.net) https://www.acmicpc.net/problem/1181

 

 

기초적인 스택의 동작 방식만 알면 쉽게 풀 수 있는 문제였다.

 

코드 

import sys
input = sys.stdin.readline

stack = []
n = int(input())
for i in range(n):
    x = list(map(int, input().split()))
    if x[0] == 1:
        stack.append(x[1])
    elif x[0] == 2 :
        if stack:
            print(stack.pop())
        else:
            print(-1)
    elif x[0] == 3:
        print(len(stack))
    elif x[0] == 4:
        if stack:
            print(0)
        else:
            print(1)
    else:
        if stack:
            print(stack[-1])
        else:
            print(-1)