[Swift] 프로그래머스 위클리 챌린지 5주차 모음 사전

Problem

 

코딩테스트 연습 - 5주차

---

Solution

 

1. 모음 순서와 인덱스별 카운트 어레이를 만든다.

 

인덱스별 카운트는 가장 끝 즉,4번째 인덱스일 경우 다음 모음으로 이동하는데 1개의 단어가 필요합니다.

 

ex) AAAAA -> AAAAE -> AAAAI .... 1개씩

 

3번째 인덱스의 경우 다음 모음으로 이동하는데 6개가 필요합니다.

 

ex) AAAA -> AAAE -> AAAI ... 6개씩

 

2번째 인덱스의 경우 다음 모음으로 이동하는데 31개가 필요합니다.

 

ex) AAA -> AAE -> AAI 31개씩

 

... 이런식으로 각 필요한 단어의 갯수를 미리 구해놓습니다.

 

 let vowelsDic = ["A":0,"E":1,"I":2,"O":3,"U":4]
 let count = [781,156,31,6,1]

 

2. 단어의 인덱스별로 모음의 순서와 인덱스별 카운트를 곱해서 모두 더해줍니다.

 

여기서 다음 문자로 이동할 때 1개씩 단어의 수가 세어지므로 단어의 갯수도 더해줍니다.

 

 return word.enumerated().map{count[$0.offset]*vowelsDic[String($0.element)]!}.reduce(word.count) {$0 + $1}

---

Source Code