파이썬 입력 받기 - sys.stdin.readline()

 input()대신 sys.stdin.readline()을 사용하는 이유

한 두줄 입력받는 문제들과 다르게, 반복문으로 여러줄을 입력 받아야 할 때는 input()으로 입력 받는다면 시간초과가 발생할 수 있음

 

 

sys.stdin.readline() 사용법

한 개의 정수를 입력받을 때

import sys
a = int(sys.stdin.readline())

 

임의의 개수의 정수를 한줄에 입력받아 리스트에 저장할 때

import sys
data = []
n = int(sys.stdin.readline())
for i in range(n):
    data.append(list(map(int,sys.stdin.readline().split())))

 

문자열 n줄을 입력받아 리스트에 저장할 때

import sys
n = int(sys.stdin.readline())
data = [sys.stdin.readline().strip() for i in range(n)]