[코딩테스트] Python 입력(Input) 방법

1. input() 사용하기

가장 기본적인 입력을 받는 방법입니다.

data = input() # 데이터를 입력 받음
print(data)

intData = int(input()) # 정수형 데이터를 입력 받음
print(intData)

data1, data2 = input().split() # 여러 데이터를 입력 받아 각 변수에 저장
print(data1, data2)

listData = list(map(int, input().split())) # 리스트 형태로 입력 데이터를 저장
print(listData)

 

 

 

2. sys.stdin.readline() 사용하기

input()은 상대적으로 속도가 느려서 백준과 같은 시스템에서 평가를 할 때 시간 초과가 나는 경우, 이를 해결하는 방법으로 사용할 수 있습니다.

import sys

data = int(sys.stdin.readline()) # 데이터를 입력 받음
print(data)

data1, data2 = sys.stdin.readline().split() # 여러 데이터를 입력 받아 각 변수에 저장
print(data1, data2)

listData = list(map(int, sys.stdin.readline().split())) # 리스트 형태로 입력 데이터를 저장
print(listData)

 

 

 

3. 파일로 만들어 입력 자동으로 처리하기

테스트 케이스가 많은 경우나, 입력 예제가 복잡한 경우, 매번 테스트 할 때마다 작성이 번거로운 경우에는 txt파일에 별도의 입력 데이터를 저장하고, 이를 사용하여 입력 값을 자동으로 들어가도록 할 수 있습니다.

# input.txt
ANGELPLAYER
1234
1 3 5 7 9

# main.py
import sys

from cv2 import line
sys.stdin = open("input.txt", "r")

line1 = input()
line2 = int(input())
line3 = list(map(int, sys.stdin.readline().split()))

print(line1) # ANGELPLAYER
print(line2) # 1234
print(line3) # [1, 3, 5, 7, 9]