열렬히.뛰기

10828번: 스택

알고리즘: 실전 > 알고리즘 강의 > 10828번: 스택

코드

python
from typing import List
import sys
def find(order, A: List):
    if 'push' in order:
        p, x = str(order).split(" ")
        A.append(int(x))
    elif 'pop' in order:
        if len(A) == 0:
            print("-1")
        else:
            print(A[len(A)-1])
            A.pop()
    elif 'size' in order:
        print(len(A))
    elif 'empty' in order:
        if len(A) == 0:
            print("1")
        else:
            print("0")
    elif 'top' in order:
        if len(A) == 0:
            print("-1")
        else:
            print(A[len(A)-1])
    return A

A = []
for _ in range(int(sys.stdin.readline())):
    A = find(sys.stdin.readline(), A)