[Python]_코딩테스트_가운데 숫자 가져오기

[프로그래머스] 가운데 숫자 가져오기

Problem

문제설명

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

제한사항

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

입출력 예

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

---

Solution

s 길이 : 짝수

if len(s) % 2 == 0:
        num = len(s) // 2
        return s[num-1:num+1]

• num : 가운데 숫자
• 짝수의 경우 가운데 두글자를 반환해야 하기 때문에 len(s) // 2 =(몫)
  값의 -1, +1
• 예) s = "abcdef" = len(s) = 6
  6//2 = 3
  s[3-1:3+1] = s[2:4]
  "c,d"

s 길이 : 홀수

else:
        num = len(s) // 2
        return s[num]

• 홀수는 중앙값이 존재하기 때문에 len(s)//2 를 한 후 그대로 슬라이싱해서 가져옴

완성 코드

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