[프로그래머스] 위장 (Lv 2) - javascript

문제

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

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.
 

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

• 제한사항
  • clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
  • 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
  • 같은 이름을 가진 의상은 존재하지 않습니다.
  • clothes의 모든 원소는 문자열로 이루어져 있습니다.
  • 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
  • 스파이는 하루에 최소 한 개의 의상은 입습니다.
     

풀이

 
해시 알고리즘을 사용하는 문제다.
주어진 배열을 종류에 따라 객체로 정리해서 요소의 개수를 세고, 최종 경우의 수를 구하면 된다.
종류마다 각 요소가 무엇인지 저장할 필요는 없다.
조합의 수만 구하면 되기 때문이다.
 
예시로 입력이 [["yellowhat", "headgear"], ["bluesunglasses", "eyewear"], ["green_turban", "headgear"]] 로 주어진다면

• headgear : yellowhat , green_turban
• eyewear : bluesunglasses
   
  종류별로 위와 같이 정리할 수 있다.
   

최종 조합의 수

=> (headgear의 종류 가지수(2) + 착용하지 않을 경우(1)) x (eyewear의 종류 가지수(2) + 착용하지 않을 경우(1)) - 1

-1을 해주는 이유는 착용하지 않을 경우로 조합이 만들어져 아무것도 착용 안한 상태가 되는 경우의 수가 있기 때문이다.
 
이제 이를 코드로 구현해보자.
 

function solution(clothes) {

    let answer = 1;

    let object = {}; // 빈 객체 생성

    for (let i = 0; i < clothes.length; i++) {
        object[clothes[i][1]] = (object[clothes[i][1]] || 1) + 1; // *
    }

    for(let key in object) {
        answer *= object[key]; // 객체의 값 곱하기
    }

    return answer - 1;
}

 

• object[clothes[i][1]] || 1
object[clothes[i][1]] 존재 X (undefined) => 1 반환
  object[clothes[i][1]] 존재 O => 값 반환
   

• 정확도 테스트 결과

  

  
   

다른 풀이

function solution(clothes) {
    return Object.values(clothes.reduce((obj, t)=> {
        obj[t[1]] = obj[t[1]] ? obj[t[1]] + 1 : 1;
        return obj; } , {})).reduce((a,b)=> a*(b+1), 1)-1; }

 

reduce를 이용한 풀이인데 가독성이 안 좋고 속도도 크게 차이나지 않아서 비추

 

참고

https://velog.io/@fastpace04/