백준 10828 - 스택 (class2)

https://www.acmicpc.net/problem/10828'

10828번: 스택

 

처음 시도에선 C언어에서 구현하듯 top포인터를 사용해 풀려다가 파이썬의 편리한 pop() 을 이용해 쉽게 풀고자 했다.

index error가 났던 것은 underflow, overflow 처리.

import sys

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

for _ in range(num):
    word = sys.stdin.readline().split()
    cmd = word[0]
    # top = -1

    if cmd == "push":
        stack.append(word[1])
        # top += 1

    elif cmd == "pop":
        # val = stack[top]
        # top -= 1
        if len(stack) == 0:
            print(-1)
        else:
            val = stack.pop()
            print(val)

    elif cmd == "size":
        # print(top + 1)
        print(len(stack))

    elif cmd == "empty":
        # if top > -1:
        #     print(0)
        # else:
        #     print(1)
        if len(stack) == 0:
            print(1)
        else:
            print(0)

    elif cmd == "top":
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])
        # print(stack[top])


    # else:
    #     print("error")