Programmers / Level 2 / 모음사전 / JS

< 문제 바로가기 >

프로그래머스

 

function solution(word) {
  let result = 0;
  const ALPHABET_LIST = ['A', 'E', 'I', 'O', 'U'];
  const COUNT_TO_CHANGE_LIST = [781, 156, 31, 6, 1];

  result = word.split('').reduce((acc, cur, idx) => {
    return acc + ALPHABET_LIST.indexOf(cur) * COUNT_TO_CHANGE_LIST.at(idx) + 1;
  }, 0);

  return result;
}

---

COUNT_TO_CHANGE_LIST 라는 변수는 각 자리가 바뀌려면 몇 번의 카운트가 지나야 바뀌는지를 나타낸 배열이다. 예를들어 "AAEOO" 라는 word에서 맨 처음 A가 E로 바뀌려면 781번이 지나야하고, 두 번째 A가 E로 바뀌려면 156번, ... , 마지막 O가 바뀌려면 한 번의 카운트만 지나면 된다.

 

이 수들은 거꾸로 생각해보면 알 수 있는데, 먼저 마지막 O가 바뀌려면 단순히 그 다음에 U로 바뀌므로 카운트만 하나 더해주면된다. 그런데 나머지는 모두 유지한채 네 번째의 O가 바뀌려면, 즉 "AAEUO"가 되려면 6번의 카운트가 지나야한다.

맨 마지막의 문자가 U, A, E, I, O 를 거쳐서 돌아오기 때문에 5번의 카운트와, "AAEOU" 다음은 "AAEU" 이므로 이 한번의 카운트를 더해주는것이다.

 

세번째가 바뀌는 경우는 위 경우를 네번째가 A, E, I, O, U 일때 5번과 똑같이 "AAI"가 되었을 때 한번을 더해주게 된다.

따라서 6 * 5 + 1 이 되어 총 31번의 카운트가 지나야한다.

 

위와 같이 두번째는 31 * 5 + 1 = 156, 첫번째는 156 * 5 + 1 = 781의 규칙성을 가지게 된다.

 

이러한 규칙을 가지고 reduce를 이용해 더해주면 되는데, 사전 순으로 A, E, I, O, U 순이기 때문에 이것을 배열에 넣고, 각 인덱스만큼 곱해줘서 + 1을 한 값을 return 해주면 된다. 인덱스의 의미는 예를들어 word의 첫번째가 I라면 위와같은 781번의 카운팅을 두번해주어야 한다. 따라서 배열에서 'I'의 인덱스인 2를 곱해주는 방식이다.