Programmers 코딩테스트 연습 LV 1 - 가운데 글자 가져오기

문제 설명

단어 s의 가운데 글자를 반환하는 함수, solution을 만들어 보세요. 단어의 길이가 짝수라면 가운데 두글자를 반환하면 됩니다.

---

재한사항

• s는 길이가 1 이상, 100이하인 스트링입니다. 

---

입출력 예

s	return
"abcde"	"c"
"qwer"	"we"

---

 

import math

def solution(s):
    answer = ''
    if len(s)%2 == 1:
        answer = s[ math.floor(len(s)/2)]
    else:
        num = len(s)//2
        answer = s[num-1:num+1]
    return answer

 

[코드 설명]

 

이번 문제는 문자열의 가운데 글자를 가져오는 문제로써, 문자열이 홀수 일 때는 중앙의 글자 하나를 가져오고 짝수일 경우에는 가운데 두 글자를 가져와야합니다.

따라서 if문을 사용하여 문자열의 길이가 홀수와 짝수일 때를 나누어 생각하였으며, 홀수의 경우 math 라이브러리의 .floor()함수를 사용하여 s문자열 길이의 절반 값을 올림한 값을 넣어 answer에 올바른 글자가 할당될 수 있게 하였고, 짝수의 경우 num이라는 변수에 문자열의 길이의 절반 값의 몫을 넣어 s[num-1:num+1] 이 조건과 같이 문자열 s를 슬라이싱 하여 answer에 올바른 글자가 할당될 수 있게 하였습니다.