[백준] 10828 : 스택

letsbebrave·2022년 4월 8일
0

codingtest

목록 보기
83/146
post-thumbnail

문제

풀이

import sys

t = int(sys.stdin.readline())
stack = []

def sol(n):
    global stack
    if "push" in n:
        stack.append(n.split()[1])
    if "pop" == n:
        if len(stack) == 0:
            return -1
        else:
            print(stack.pop()) # pop이 잘 안됐었음
    if "size" in n:
        return len(stack)
    if "empty" in n:
        if len(stack) > 0:
            return 0
        else:
            return 1
    if "top" in n:
        if len(stack) > 0:
            return stack[-1]
        else:
            return -1

for i in range(t):
    n = sys.stdin.readline().strip()
    if sol(n) != None:
        print(sol(n))
profile
그게, 할 수 있다고 믿어야 해

0개의 댓글