보통 input()을 사용해 입력을 받는데 반복문을 통해 입력이 많아질 때 시간 초과가 일어날 수 있다.
import sys
# 한 개 입력
a = int(sys.stdin.readline())
# 여러개 입력
a, b = map(int, sys.stdin.readline().split())
# 1차원 배열
a = list(map(int,sys.stdin.readline().split()))
# 2차원 배열
a = []
n = int(sys.stdin.readline())
for i in range(n):
a.append(list(map(int,sys.stdin.readline().split())))