[algorithm] 백준 10828 - 스택 (파이썬)

📚 문제

입력

출력

예제 입력 1

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

예제 출력 1

2
2
0
2
1
-1
0
1
-1
0
3

예제 입력 2

7
pop
top
push 123
top
pop
top
pop

예제 출력 2

-1
-1
123
123
-1
-1

🧑🏻‍💻 풀이 과정

• n번동안 반복하며 명령어(command)를 입력받자.
• size면 stack의 길이를 출력하자.
• empty면 stack이 []와 같다면 1 출력, 아니면 0을 출력하자.
• top이면 stack이 []와 같다면 -1 출력, 아니면 stack의 마지막 요소를 출력하자.
• pop이면 stack이 []와 같다면 -1 출력, 아니면 stack.pop()을 출력하자.
• push면 명령어의 5번째 인덱스부터 stack에 저장하자.

import sys


def process(stack):
    command = sys.stdin.readline().rstrip()

    # size
    if command == 'size': print(len(stack))

    # empty
    if command == 'empty': print(1) if stack == [] else print(0)

    # top
    if command == 'top': print(-1) if stack == [] else print(stack[-1])

    # pop
    if command == 'pop': print(-1) if stack == [] else print(stack.pop())

    # push
    if 'push' in command: stack.append(command[5:])


ans = []
for _ in range(int(sys.stdin.readline().rstrip())):
    process(ans)