[programmers/js, py] 모음사전

모음사전

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 이하입니다.

풀이

중복 순열을 구해주는 product 사용

from itertools import product
def solution(word):
    arr = []
    for i in range(1, 6):
        for j in list(product(["A", "E", "I", "O", "U"], repeat = i)):
            arr.append("".join(j))
    arr.sort()
    return arr.index(word)+1

DFS 사용

function solution(word) {

    const result = [];
    
    const DFS = (str, length) => {
        const arr = [..."AEIOU"];
        if (str.length === length) {
            result.push(str);
            return;
        }
        
        arr.forEach((value) => {
            DFS(str + value, length)
        })
    }
    
    const str = "";
    for (let i = 1; i <= 5; i++) DFS(str, i); // 모든 조합 수
    return result.sort().indexOf(word) + 1;
    
}