[백준][10828] 스택 | 파이썬

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

👉 문제 분석

스택에서 push, pop, size, empty, top을 구현하는 문제

 

👉 알고리즘 설계

push함수는 append를 사용하여 데이터 추가

stack의 길이로 size 출력

stack[-1]로 스택의 가장 위에 있는 정수를 출력

 

👉 코드

import sys

stack = []

x = int(sys.stdin.readline())
for i in range(x):
    y = sys.stdin.readline().split()
    if y[0] == "push":
        stack.append(y[1])
    
    elif y[0] == "pop":
        if len(stack) == 0:
            print(-1)
        else:
            print(stack.pop())

    elif y[0] =="size":
        print(len(stack))

    elif y[0] == "empty":
        if len(stack) != 0:
            print(0)
        else:
            print(1)

    elif y[0] == "top":
        if len(stack) != 0:
            print(stack[-1])
        else:
            print(-1)

 

👉 틀린 이유

런타임 에러

 

👉  틀린 부분 수정 or 다른 풀이

런타임 에러로 인해, input()을 sys.stdin.readline()로 바꿔 주었고, 여전히 런타임 에러가 나서

top부분을 구현할때, stack.top()대신 stack[-1]으로 하였더니 런타임 에러가 해결되었다.

 

👉 느낀점 or 기억할정보

시간단축을 위해 input()대신 sys.stdin.readline()을 사용