백준 10828 "스택" 파이썬 문제풀이

지난학기 수강했던 자료구조 강의에서 배운 내용을 이용하여 코드를 작성해보았다

stack = []
high = -1

def push(num) :
    global high
    stack.append(num)
    high += 1

def top() :
    if len(stack) == 0 :
        print(-1)
    else :
        print(stack[high])

def userDefPop() :
    global high
    if high == -1 :
        print(-1)
    else :
        print(stack.pop())
        high -= 1


n = int(input())


for _ in range(n):
    command = input()

    if "push" in command :
        command = command.split()
        push(command[1])

    if command == "top" :
        top()

    if command == "size" :
        print(len(stack))

    if command == "pop" :
        userDefPop()

    if command == "empty" :
        print(1) if high == -1 else print(0)

 

예제 그대로 입력해보고 출력이 동일하게 나오는 것을 확인한 후 제출하였으나, 시간 초과라는 결과와 함께 실패했다.

 

질문 게시판을 둘러보다가 input() 형식으로만 사용하면 시간초과가 날 수 있다는 글을 보게 되었다.

 

 

관련하여 검색해보니, 입출력 속도를 빠르게 하기 위해서는 

import sys
input = sys.stdin.readline

이 코드를 써주는 것이 좋다고 한다.

다만 유의해야 할 점은 위 코드가 적용된 input은 개행문자까지 읽어들이기 때문에 .rstrip()등을 이용하여 개행문자를 지워주어야 한다는 것이다.

 

 

아래는 수정된 코드이다

import sys
input = sys.stdin.readline

stack = []
high = -1

def push(num) :
    global high
    stack.append(num)
    high += 1

def top() :
    if len(stack) == 0 :
        print(-1)
    else :
        print(stack[high])

def userDefPop() :
    global high
    if high == -1 :
        print(-1)
    else :
        print(stack.pop())
        high -= 1


n = int(input().rstrip())


for _ in range(n):
    command = input().rstrip()

    if "push" in command :
        command = command.split()
        push(command[1])

    if command == "top" :
        top()

    if command == "size" :
        print(len(stack))

    if command == "pop" :
        userDefPop()

    if command == "empty" :
        print(1) if high == -1 else print(0)

 

 

 

성공!

 

 

 

---

참고

https://urakasumi.tistory.com/273

[ Python ] 입출력(I/O) 속도 빠르게 하기