[algorithm] 프로그래머스 - 가운데 글자 가져오기 (파이썬)

📚 문제

제한 사항

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

입력 예시 1

"abcde"

출력 예시 1

"c"

입력 예시 2

"qwer"

출력 예시 2

"we"

🧑🏻‍💻 풀이 과정

• 입력받은 s의 길이를 구하고 반으로 나눈다.
• 홀수라면 s[mid]를 반환하고, 짝수라면 s[mid-1]과 s[mid-1]를 반환하자.
  • 예: 길이가 5라면 01234, 길이가 6이라면 012345이므로 -1을 포함해야한다.

def solution(s):
    mid = len(s) // 2
    if len(s) % 2 == 1:
        return s[mid]
    else:
        return s[mid-1] + s[mid]


# 테스트를 위한 코드입니다.
s = "abcde"
print(solution(s))

s = "qwer"
print(solution(s))