백준 - 10828. 스택 파이썬

10828번: 스택

1. SOLUTION

스택 구현 기본 문제! 스택 모듈이 따로 존재하지 않으므로 리스트를 활용해 스택을 구현하면 된다.

sys.stdin.readline() 이거 모를 때 풀었던 문제라 input() 사용하다가 시간초과 났었음..

이 문제 풀고부터는 시간 초과 날 것 같다? 바로 sys부터 쓰고 보게 됐다.

여러모로 알고리즘 초보였던 나에게 많은 도움이 되었던 문제였다.

2. CODE

import sys
N = int(sys.stdin.readline())
stack = []

for i in range(N):
    command = sys.stdin.readline().split()
    if command[0] == 'push':
        stack.append(command[1])
    elif command[0] == 'pop':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop())
    elif command[0] == 'size':
        print(len(stack))
    elif command[0] == 'empty':
        if len(stack) == 0:
            print(1)
        else:
            print(0)
    elif command[0] == 'top':
        if len(stack) == 0:
            print(-1)
        else:
            print(stack[-1])
    else:
        continue