백준 파이썬 10828번 : 스택 시간초과 해결

10828번: 스택

스택을 몰라도 충분히 풀 수 있다. solved.ac에서 실버4 판정을 받고 있는데, 절대 그럴만한 문제는 아니다.

다만 정답률이  상당히 낮은데, 시간이 빡빡해서 그렇다. 별다른 궁리가 필요하지는 않고, input대신에 값을 입력받는 데 더 빠른 readline을 쓰면 된다.

백준 문제에서 시간초과가 날 때는 대부분 readline을 쓰거나 PyPy로 제출하면 해결된다.

 

참고로

import sys

input = sys.stdin.readline 

을 입력해두면 input 함수를 사용하는 것처럼 입력하면서 readline을 이용할 수 있다. 

 

import sys
input = sys.stdin.readline
n = int(input())
stk = [] 
for _ in range(n):
    x = list(map(str,input().split()))
    if x[0] =='push':
        stk.append(int(x[1]))
    elif x[0] =='pop':
        if stk:
            print(stk.pop())
        else:
            print('-1')
    elif x[0] =='size':
        print(len(stk))
    elif x[0] =='empty':
        if stk:
            print('0')
        else:
            print('1')
    elif x[0] =='top':
        if stk:
            print(stk[-1])
        else:
            print('-1')