[Python] input, sys.stdin.readline 차이점

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
graph = []
for i in range(n):
    graph.append(list(map(int, input().rstrip())))

빠르게 입력 받기 위해 input보다는 sys.stdin.readline을 사용한다.

 

graph 입력을 아래와 같은 코드로 입력 받았다.

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
graph = []
for i in range(n):
    graph.append(list(map(int, input())))

코드를 실행한 후

3 3

010 을 입력하자

ValueError: invalid literal for int() with base 10: '\n'오류가 발생했다.

 

📌input  과  sys.stdin.readline

input : 입력에 rstrip() 후 리턴

sys.stdin.readline : 개행문자(\n)를 포함한 값 리턴

 

따라서 아래와 같은 코드로 바꿔줘야 한다.

 

import sys

input = sys.stdin.readline

n,m = map(int, input().split())
graph = []
for i in range(n):
    graph.append(list(map(int, input().rstrip())))