python으로 코딩테스트를 준비하다보면 항상 입력값이 주어지고 그에 대한 변수를 만들어 입력값을 변수에 할당하는 것 부터 시작된다.
먼저 map과 split에 대해 알아보겠다.
map(function, iterable)
첫번째 인자는 함수
두번째 인자는 반복가능한 자료형
map 함수의 반환 값은 map 객체이기 때문에 사용하기 위해서는 list 혹은 tuple로 만들어줘야한다.
test_list = ["1", "2", "3", "4"] test = list(map(int, test_list)) print(test)
[1, 2, 3, 4]
split(sep='구분자', maxsplit=분할횟수)
split함수는 문자열을 나눠서 리스트로 만들어준다.
split()
의 default 는 띄어쓰기이다.
word = "hi hi yi yi hi yi" print(word.split()) print(word.split(" ", 2))
['hi', 'hi', 'yi', 'yi', 'hi', 'yi'] ['hi', 'hi', 'yi yi hi yi']
ex)
정수 3개가 공백을 두고 입력된다면 다음과 같이 정수3개를 변수에 할당할 수 있다.
num = list(map(int,input().split())) #10 100 1000 입력 print(num)
[10, 100, 1000]
문자열을 할당시켜야할경우는 split만으로 할당가능하다
ex)
공백이 하나인 문자열 입력
word1, word2 = input().split() # hi yi 입력 print(word) print(num)
hi yi