[Dev] 코딩 테스트에서 자주 쓰는 입력방식에 대해 (input, split, map)

코딩테스트 문제를 풀다 보면 첫 번째 단계로 사용자에게 데이터를 입력받는 경우가 매우 많습니다
눈을 감고 작성할 수 있을 때까지 연습해 둡시다

---

input()

input() 함수는 한 줄의 문자열(str)을 입력받는 함수입니다

num = int(input("정수를 입력하세요: ")) 	# input() 으로 입력받은 데이터의 기본 자료형은 문자열
print("입력받은 정수:", num)

---

 

문자열.split(separator, maxsplit)

split() 함수는 문자열을 separator 기준으로 나누어 리스트로 반환하는 메서드입니다

sentence = "potato tomato onion"
print(sentence .split())

• split() 함수를 파라미터 없이 사용하면 공백기준으로 문자열을 나누어 리스트로 반환합니다

공백으로 문자열을 나눔

sentence = "potato,tomato,onion"
print(sentence.split(','))

• split() 함수의 첫 번째 파라미터(separator)를 기준으로 문자열을 나누어 리스트로 반환합니다

, 를 기준으로 문자열을 나눔

---

map()

map() 함수는 리스트의 모든 원소에 각각 특정한 함수를 적용할 때 사용합니다

a = list(map(str, range(10)))
print(a)

• range(10)에서 반환된 정수형 리스트의 모든 원소에 str 함수를 적용하여 모두 문자열로 변환합니다

리스트의 모든 원소에 str 함수가 적용

 

---

전형적인 입력 방법

이제 함수들을 활용하여 코딩테스트에서 자주 나오는 입력 방식을 구현할 수 있습니다

Case 1. 한 번에 변수 데이터를 입력받는 경우

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

a, b, c

Case 2. 한 번에 리스트 데이터를 입력받는 경우

array = list(map(int, input().split()))
print(array)

array