[BOJ/백준] 10828번 스택 | 파이썬 (🌠sys.stdin.readline으로 시간 초과 해결)

문제

풀이

import sys

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

 

구현 자체는 어렵지 않았지만 input()을 사용했다가 시간 초과가 나서 당황했던 문제.

 

input() 대신 sys.std.readline()을 사용해서 해결했다.

 

🌠input() vs sys.stdin.readline()

input([prompt])은 파이썬 내장함수다. prompt argument가 있을 때 argument는 줄 바꿈 문자 없이 출력된다. 그 후 input() 함수는 입력으로부터 한 줄을 읽고, 문자열로 변환한 후 반환한다. 줄 끝의 줄 바꿈 문자는 제거한다.

 

sys.stdin.readline()은 file object로 분류된다. 입력을 읽기 위해 다른 파일 객체를 만드는 것과 같다. input()과 달리 prompt argument를 받아 출력하지 않고, 줄 바꿈 문자를 제거하는 과정이 없다. 따라서 속도가 더 빠르다.

 

sys.stdin.readline()은 sys 모듈을 import 해서 사용한다.

* sys 모듈을 사용하면 파이썬 인터프리터가 제공하는 변수와 함수를 직접 제어 가능

 

---

풀이에서는 입력값의 개수 n을 맨 처음 입력받기 위해 int(sys.stdin.readline())을 사용했다. 형 변환을 해주지 않으면 줄 바꿈 문자를 포함한 값이 저장되기 때문에 주의해야 한다.

 

여러 줄의 명령을 입력받아야 하는 반복문 for에서는 sys.stdin.readline().rstrip()를 사용해 오른쪽 공백을 제거한 후 값을 입력받았다.

---

백준 18828번 : 스택

10828번: 스택

파이썬 공식문서

Built-in Functions — Python 3.9.6 documentation

input() vs sys.stdin.readline()

[Python] Input vs. sys.stdin.readline 차이점?

sys.stdin.readline() 사용법

[Python 문법] 파이썬 입력 받기(sys.stdin.readline)