[10828] 알고리즘 - 스택

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

백준 10828번을 풀어보자!

10828번: 스택

 

1. 첫시도 (틀림. 시간초과)

co_num=0

co_num=input()
co_num=int(co_num)

stack = []

for i in range(co_num):

    b=input()
    
    if "push" in b:
        stack.append(int(b[5:]))
    elif "pop" in b:
        if stack:
            print(stack[-1])
            stack=stack[:-1]
        else:
            print("-1")
    elif "size" in b:
        print(len(stack))
    elif "empty" in b:
        if stack:
            print(0)
        else:
            print(1)
    elif "top" in b:
        if stack:
            print(stack[-1])
        else:
            print(-1)

 

 

결과는 잘 나왔는데 백준에서는 런타임 에러가 떴다.

 

2. 두 번째 시도 (성공)

import sys

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

for i in range(co_num):

    b = sys.stdin.readline().split() #input 대신 sys.stdin.readline()사용
    ch=b[0]
    
    if "push" in ch:
        stack.append(int(b[1]))
    elif ch=="pop":
        if stack:
            print(stack[-1])
            stack=stack[:-1]
        else:
            print(-1)
    elif ch=="size":
        print(len(stack))
    elif ch=="empty":
        if stack:
            print(0)
        else:
            print(1)
    elif ch=="top":
        if stack:
            print(stack[-1])
        else:
            print(-1)

input대신 sys.stdin.readline() 사용하니까 성공했다. 백준에서는 입력받을 때 input사용하면 시간초과 에러가 자주 뜬다고 한다.

 

readline과 input 차이점

readline은 마지막 개행문자 \n도 저장해서, input보다 길이가 한 글자 더 길다.