가오리의 코딩일기

[10828] 스택 본문

Python/백준

[10828] 스택

류경혜 2022. 5. 29. 13:30

import sys
N = int(sys.stdin.readline())
# N = int(input())
stack = []
for _ in range(N):
    # word = input().split()
    word = sys.stdin.readline().split()
    order = word[0]
    if order == "push":
        value = word[1]
        stack.append(value)
    elif order == "pop":
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop())
    elif order == "size":
        print(len(stack))
    elif order == "empty":
        if len(stack) == 0:
            print(1)
        else:
            print(0)
    elif order == "top":
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])

'Python > 백준' 카테고리의 다른 글

[2751] 수 정렬하기2  (0) 2022.05.29
[9012] 괄호  (0) 2022.05.29
[11004] k번째 수  (0) 2022.05.29
[11652] 카드  (0) 2022.05.29
[10989] 수 정렬하기3  (0) 2022.05.29