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()은 반복 가능한 객체(리스트 등)에 대해 각각의 요소들을 지정된 함수로 처리해주는 함수입니다.