[Python] BOJ 10828번. 스택

작성일 | In |

10828번. 스택

문제 링크

풀이 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# 10828번. 스택

# 시간초과 방지용 readline
import sys

input = sys.stdin.readline

n = int(input())
stack = []
for i in range(n):
    s = input().strip()
    if s.split()[0] == 'push':
        stack.append(int(s.split()[1]))
    elif s == 'pop':
        if not stack:
            print(-1)
        else:
            print(stack.pop())
    elif s == 'size':
        print(len(stack))
    elif s == 'empty':
        if not stack:
            print(1)
        else:
            print(0)
    elif s == 'top':
        if not stack:
            print(-1)
        else:
            print(stack[-1])

비고