[iOS Swift] Lv.1 가운데 글자 가져오기

출처: 프로그래머스
https://school.programmers.co.kr/learn/courses/30/lessons/12903
난이도:Lv.1

 

문제 설명

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

 

제한 사항

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

입출력 예

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

---

func solution(_ s:String) -> String {
    // s의 길이가 짝수일때와 홀수일때를 나눠서 보자구
    if s.count % 2 == 0{
      let AA = s.index(s.startIndex, offsetBy: (s.count / 2) - 1)
      let BB = s.index(s.startIndex, offsetBy: s.count / 2)
      return String(s[AA...BB])  
    }else{
      let CC = s.index(s.startIndex, offsetBy: s.count / 2)
      return String(s[CC])
      }
    
}