가운데 문자 추출하기

문제

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

재한사항

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

입출력 예

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

코드

def solution(s):
    answer = ''
    length = len(s) # 문자열 길이
    
    if length == 1: # 문자일 때
        answer = s
    elif length % 2 == 1: # 문자열 길이가 홀수일 때
        answer = s[length//2]
    else : # 문자열 길이가 짝수일 때
        answer = s[length//2-1]
        answer = answer + s[length//2]
    
    return answer