[Algorithm] 입력 받는 방법?

코딩테스트를 준비하다보면 제일 처음으로 하는 것이 주여진 정보를 입력받는 것이다. 따라서 이번에는 파이썬을 이용하여 정보를 입력받는 방법에 대해서 다뤄보겠다.

 

다음은 공백을 기준으로 2개의 변수를 입력 받는 방법이다. 

 

n, m = map(int, input().split())

[3 5] 이렇게 정보가 입력된다면 n=3, m=5 정보가 저장되게 된다.

 

다음은 리스트를 입력 받는 방법이다.

 

data = list(map(int, input().split()))

 

1 2 3 4 5 를 입력하게 된다면 data에는 [1,2,3,4,5]가 저장되게 된다.

 

다음은 크기가 n인 배열을 입력 받는 방법이다.

 

n = int(input())
 
array = []
for i in range(n):
  array.append(int(input()))

 

3

5

4

3 이렇게 정보가 저장되게 되면 array에는 [5,4,3]이 저장되게 된다.

 

import sys
data = sys.stdin.readline().rstrip()

 

입력의 개수가 많은 경우에는 파이썬 라이브러리인 sys를 사용한다.  해당 라이브러리를 사용하면 한 줄씩 입력을 처리할 수 잇어서 빠르게 입력받을 수 있다.