[백준] 10828번 문제 (스택) 파이썬(Python) 풀이

문제 내용

 

import sys
stack = []

for _ in range(int(sys.stdin.readline())):
    cmd = sys.stdin.readline().split()

    if len(cmd) == 2:
        stack.append(cmd[1])
    elif cmd[0] == 'top':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])
    elif cmd[0] == 'size':
        print(len(stack))
    elif cmd[0] == 'empty':
        print(1 if len(stack) == 0 else 0)
    elif cmd[0] == 'pop':
        if len(stack) == 0:
            print(-1)
        else:
            data = stack[-1]
            del stack[-1]
            print(data)

 

스택을 알고 있다면 아주 간단하게 풀 수 있는 문제다.

예외만 잘 처리해주면 된다.

 

단 파이썬을 사용할 경우 시간 초과에 주의해줘야 된다..