[Swift] lv.1 가운데 글자 가져오기 | Programmers

문제 설명

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

제한 사항

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

1차 고민

홀수와 짝수일 경우를 가장 먼저 생각해야한다. if와 else로 나누는 방법이 제일 깔끔할 것 같고
배열로 바꿔서 가운데 배열을 출력하면 되는 문제인 것 같다.
5의 경우 2.5니까 2로 인식될거고 0 1 2 3 4 중에서 2 출력이 맞고
4의 경우 2니까 0 1 2 3 중에서 나누기 2에다가 -1 먼저 출력 + 나누기 2 출력 하면 1 2 두 개만 출력 가능.

1차 코드

func solution(_ s:String) -> String {

    if s.count % 2 == 1 {
        return String(Array(s)[s.count/2])
    } else {
        return String([Array(s)[s.count/2-1],Array(s)[s.count/2]])
    }
}

1차 결과