알고리즘 - 프로그래머스 가운데 글자 가져오기 (Swift)

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

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

 

# 홀수, 짝수 일때 가져오는 인덱스가 다르다.

 

아래 방식으로 풀기는 했지만...;; forEach 안쓰고 푸는 방법으로 다시 풀어보자!

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

    var result = String()
    s.enumerated().forEach { (index, char ) in

        if s.count % 2 == 0  {
            if index == (s.count / 2) - 1 || index == (s.count / 2)  {
                result = result + String(char)
            }
        } else {
            if index == s.count / 2 {
                result = result + String(char)
            }
        }
    }

    return result
}