greentea31's Note

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

풀이 과정

파이썬에서는 리스트를 스택처럼 이용할 수 있다. push 명령이 들어오면 list 끝에 붙이고, pop 명령이 들어오면 list 끝의 원소를 출력한뒤 삭제하면 된다. empty 판단은 len() 함수를 이용해서 길이가 0이면 비었다고 판단하면 된다.

이 문제는 입력의 양이 많은 문제로 input() 명령으로 입력을 받으면 시간 초과가 발생할 수 있다. input()보다 빠르게 입력을 받기 위해 sys.stdin.readline() 명령을 사용하였다.

코드

import sys

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

for i in range(N):
    prompt = sys.stdin.readline().rstrip()
    if prompt == 'top':
        if len(stack) == 0: print(-1)
        else: print(stack[-1])
    elif prompt == 'pop':
        if len(stack) == 0: print(-1)
        else:
            print(stack[-1])
            del stack[-1]
    elif prompt == 'size':
        print(len(stack))
    elif prompt == 'empty':
        print(1 if len(stack) == 0 else 0)
    elif prompt[:4] == 'push':
        stack.append(int(prompt[5:]))