본문 바로가기

알고리즘/프로그래머스

[JS] Lv.2 위장

반응형

안녕하세요. pitang입니다.

 

위장

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수 작성합니다.

 

[제한 조건]

- clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.

- 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.

- 같은 이름을 가진 의상은 존재하지 않습니다.

- clothes의 모든 원소는 문자열로 이루어져 있습니다.

- 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '-'로만 이루어져 있습니다.

- 스파이는 하루에 최소 한 개의 의상은 입습니다.

 

[예시]

clotehs return
[["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]] 5
[["crow_mask", "face"], ["blue_sunglasses", "face"], ["smoky_makeup", "face"]] 3

 

[정답]

function solution(clothes) {
    // 옷을 입지 않은 경우의 수도 있음.
    
    let answer = 1;
    
    let types = {};
    const clothesLen = clothes.length;
    
    for(let i = 0; i < clothesLen; i++) {
        if(!types[clothes[i][1]]) {
            types[clothes[i][1]] = 1;
        } else {
            types[clothes[i][1]]++;
        }
    }
    
    for(part of Object.values(types)) {
        answer *= (part+1);
    }
    
    return answer - 1;
}

- 의상의 종류 이름을 types의 저장하고 개수를 value로 저장해 주었다.

- 의상의 개수에 1을 더해주고(왜? 옷을 입지 않은 경우의 수도 있기에) 모두 곱해준 다음,

   1을 빼(왜? 아무것도 입지 않은 것은 빼야 하므로) answer 변수를 return

 

감사합니다.

728x90
반응형