[백준/Python] 10828번 스택

알고리즘/백준

[백준/Python] 10828번 스택

ㅈㅣ니 2024. 2. 13. 16:34

https://www.acmicpc.net/problem/10828

10828번: 스택

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지

www.acmicpc.net

import sys

t = int(input())
stack = []
def push(num):
    stack.append(num)

def pop():
    if len(stack) == 0:
        print(-1)
    else:
        print(stack[-1])
        stack.pop()

def size():
    print(len(stack))

def empty():
    if len(stack) == 0:
        print(1)
    else :
        print(0)

def top():
    if len(stack) == 0:
        print(-1)
    else :
        print(stack[-1])

for i in range(t):
    
    msg = sys.stdin.readline().split()

    if msg[0] == 'push':
        push(int(msg[1]))
    elif msg[0] == 'top':
        top()
    elif msg[0] == 'size':
        size()
    elif msg[0] == 'empty':
        empty()
    elif msg[0] == 'pop':
        pop()