[Python3] input 입력 받기

백준, 알고스팟 같은 알고리즘 사이트 솔루션을 작성할 때 테스트 케이스에 대한 입력을 감안하여 솔루션을 작성해야 합니다. 이 글에선 솔루션 작성 때 자주 사용하는 input 3가지 경우를 소개합니다.

 

1개의 값 입력받기

input 함수를 사용하여 1개의 값을 입력받을 수 있습니다. 입력 받은 값의 자료형은 문자열입니다.

a = input()

 

N개의 값 입력받기

input 함수 뒤에 split 함수를 작성하여 구현할 수 있습니다. 

b,c = input().split()
d = input().split()

 

 

리스트로 입력받기

반복문을 활용하여 입력받는 값을 원하는 자료형으로 만들 수 있습니다. 아래 코드는 리스트로 값는 예제입니다.

team = []
for i in range(3):
    b,c = input().split()
    type(b)
    team_mate = [b,c]
    team.append(team_mate)
print(team)

# 1 2
# 2 3
# 4 5
# [['1', '2'], ['2', '3'], ['4', '5']]

 

split() 후 자료형은 문자열

split() 함수를 사용하여 나온 각 원소들의 자료형은 문자열입니다. 이후 변환을 하는 작업을 줄이기 위해 map 함수를 사용할 수도 있습니다.

b,c = map(int, input().split())
print(type(b),c)

# <class 'int'> 2