본문 바로가기

알고리즘 삽질장

[프로그램스] 가운데 글자 가져오기

반응형


문제설명

https://programmers.co.kr/learn/courses/30/lessons/12903

 

코딩테스트 연습 - 가운데 글자 가져오기

단어 s의 가운데 글자를 반환하는 함수, solution을 만들어 보세요. 단어의 길이가 짝수라면 가운데 두글자를 반환하면 됩니다. 재한사항 s는 길이가 1 이상, 100이하인 스트링입니다. 입출력 예 s ret

programmers.co.kr

사고과정

  • 홀수 일 경우는 문자길이를 2로 나누었을 때의 인덱스를 가져오면 된다.
  • 짝수 일 경우는 문자길이를 2로 나누었을 때의 인덱스 -1 부터 +1 인덱스까지의 슬라이싱으로 가져오면 된다.

풀이

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