[프로그래머스 / Python] 가운데 글자 가져오기 | 초코더

문제 설명

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

제한사항

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

입출력 예

s	return
abcde	c
qwer	we

접근방법

우선 홀수일 때와 짝수일 때의 경우를 나눠주어야 합니다.

그리고 string 형태도 배열처럼 s[1] 이런식으로 사용가능 하다는 점을 알고 있어야 합니다.

나의코드

def solution(s):
    #홀수
    if len(s)%2 == 1:
        return s[len(s)//2]
    #짝수
    else :
        return s[(len(s)//2)-1:(len(s)//2)+1]