[프로그래머스 Level.2] 모음사전 (완전탐색) (Java)

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/84512

프로그래머스

 

 

 

코딩테스트 연습 > 완전탐색 > 모음사전

 

 

 

문제 설명

 

 

 

사전에 알파벳 모음 'A', 'E', 'I', 'O', 'U'만을 사용하여 만들 수 있는, 길이 5 이하의 모든 단어가 수록되어 있습니다. 사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA"이며, 마지막 단어는 "UUUUU"입니다.

 

단어 하나 word가 매개변수로 주어질 때, 이 단어가 사전에서 몇 번째 단어인지 return 하도록 solution 함수를 완성해주세요.

 

 

 

제한사항

• word의 길이는 1 이상 5 이하입니다.
• word는 알파벳 대문자 'A', 'E', 'I', 'O', 'U'로만 이루어져 있습니다.

 

---

 

입출력 예

word	result
"AAAAE"	6
"AAAE"	10
"I"	1563
"EIO"	1189

 

 

입출력 예 설명

 

입출력 예 #1

 

사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA", "AAA", "AAAA", "AAAAA", "AAAAE", ... 와 같습니다. "AAAAE"는 사전에서 6번째 단어입니다.

 

 

 

입출력 예 #2

 

"AAAE"는 "A", "AA", "AAA", "AAAA", "AAAAA", "AAAAE", "AAAAI", "AAAAO", "AAAAU"의 다음인 10번째 단어입니다.

 

 

 

입출력 예 #3

 

"I"는 1563번째 단어입니다.

 

 

 

입출력 예 #4

 

"EIO"는 1189번째 단어입니다.

 

 

 

나의 코드

class Solution {
    public int solution(String word) {
        int answer = word.length();
        String base = "AEIOU";
        int[] indexNum = {781, 156, 31, 6, 1};
        
        // word의 첫문자부터 base에서 인덱스를 찾아 indexNum 수와 곱하여 더한다.
        for(int i=0; i<word.length(); i++) {
            answer += indexNum[i] * base.indexOf(word.charAt(i));
        }
        
        return answer;
    }
}

 

풀이

1. 이 문제는 규칙을 찾아서 문제를 풀었다.
2. A -> AA -> ... -> AAAAA 로 문자가 증가하고 문자는 AEIOU 순서로 바뀌는 것을 이용하여 규칙을 구하였다. 
3. 5번 째 자리가 바뀌는 경우는 AAAAA (5번 째) -> AAAAE (6번 째) 이므로 1씩 증가한다. (0 + 5^0 = 1)
   4번 째 자리가 바뀌는 경우는 AAAA (4번 째) -> AAAE (10번 째) 이므로 6씩 증가한다. (1 + 5^1 = 6)
   3번 째 자리가 바뀌는 경우는 AAA (3번 째) -> AAE (34번 째) 이므로 31씩 증가한다. (6 + 5^2 = 34)
   (3번 째 자리를 보면 AAAE (10번 째) -> AAAU (10 + 6*3 = 28번 째) -> AAE (28 + 6 = 34번 째) 인 것을 볼 수 있다.)
   2번 째 자리가 바뀌는 경우는 31 + 5^3 = 156씩 증가된다는 것을 구할 수 있고
   1번 째 자리가 바뀌는 경우는 156 + 5^4 = 781씩 증가된다는 것을 구할 수 있다.
4. 이렇게 구한 5자리가 각각 바뀌기 위해 증가하는 경우를 길이가 5인 int 배열을 생성해서 1번 째 자리가 바뀌는 경우부터 순서대로 넣어준다. 
5. 이렇게 index의 자리의 문자가 바뀌는 차례를 구했으니 word의 길이만큼 for문을 돌면서 word의 첫 번째 자리부터 "AEIOU"에서 해당 문자의 index를 구해 아까 구한 배열의 해당 자리가 바뀌는 경우와 곱하여 answer에 더해주면 된다. 그리고 word의 각 자리수가 바뀌는 경우를 구해 더해주었지 word의 길이만큼 자리수가 될 때까지 경우는 구하지 않았기 때문에 word의 길이만큼 더해주면된다. ex) word가 "AAAE"로 길이가 4라면 A -> AA -> AAA -> AAAA 가 되도록 4를 더해준다.