프로그래머스 / 가운데 글자 가져오기

Question

문제링크

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

제한사항

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

Input & Output

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

Logic

1. 길이가 1이라면 그대로 반환
2. 길이가 홀수라면 중간 위지 반환
3. 길이가 짝수라면 가운데 두 글자 반환

Code

def solution(s):
    answer = ''
    a = len(s)
    if a==1:
        return s
    elif a%2 == 1:
        answer = s[a//2]
    else :
        answer = s[a//2-1]+s[a//2]
    return answer