input() 함수로 사용자가 어떤 값을 입력하게 하고, 그 값을 변수에 저장할 수 있습니다.
반복문으로 여러줄을 입력 받아야 할 때는 input()으로 입력 받는 다면 시간초과가 발생할 수 있다. 그 때 sys.stdin.readline()을 사용해야 시간초과가 발생하지 않습니다.
from sys import stdin
# '3'을 입력했을때
a = sys.stdin.readline() # '3\n'
b = int(sys.stdin.readline() # 3
❗참고
sys.stdin.readline()은 한줄 단위로 입력받기 때문에, 개행문자가 같이 입력 받아집니다. 만약 3
을 입력했다면, 3\n
이 저장되기 때문에, 개행문자를 제거해야 합니다.
또한, 변수 타입이 문자열 형태(str)로 저장되기 때문에, 정수로 사용하기 위해서 형변환을 거쳐야 합니다.
from sys import stdin
# '1 2 3'을 입력
a, b, c = map(int, sys.stdin.readline().split()) # a = 1, b = 2, c = 3
❗참고
map()
은 반복 가능한 객체(리스트 등)에 대해 각각의 요소들을 지정된 함수로 처리해주는 함수입니다.