[프로그래머스] 위장

해달·2022년 1월 7일

문제이해

주어진 배열은 [의상의 이름, 의상의 종류] 로 이루어져있다
같은 이름을 가진 의상은 존재하지 않는다.
스파이는 하루에 최고 한 개의 의상은 입습니다.
(아무것도 안입는 경우 X)

[ ["yellowhat", "headgear"], 
  ["bluesunglasses", "eyewear"], 
  ["green_turban", "headgear"] ]

입출력 예시 중 첫번째 배열로 예시를 들어보면
headgear의 수는 A(2) eyewear의 수는 B(1) 라고 가정했을 때
headgear 와 eyewear 의 조합의 경우의 수는 A * B 이다.

이때 A만 착용하고 B를 착용하지 않을 수도 있고,
반대로 B는 착용하고 A를 착용하지 않을 수도 있다.

선택하지 않을 수도 있다는 경우의 수도 추가되기 때문에 주어진 종류의 개수에 1씩 더해준다
(A+1) * (B+1)
그렇게 되면 A와 B를 둘다 착용하지 않는 경우의 수도 1개 추가 된다

주어진 조건에서 스파이는 하루에 최소 한 개의 의상은 입습니다 라는 조건에 의하여
아무것도 입지 않는 경우의 수를 -1 로 빼주어야 한다.

최종 : (A+1) * (B+1) - 1

코드작성

function solution(clothes) {
  let result = 1;
  let obj = {};
  
  clothes.forEach((el) => {
   obj[el[1]] = (obj[el[1]] || 1) +1;
    //[의상의종류] = 개수 더해주기
  })
    
  for(let key in obj){
    result *= obj[key]
  }
    return result-1
}

obj에 처음 값이 추가 될 경우에 주어진 값이 없어 undifined가 들어가게 될 경우에 or 문을 사용하면 1로 선택이 된다
초기값이면 첫번째 경우의 수이기때문에 + 1 (안입는경우) 를 더해준다
만약 이미 값이 할당되어있다면 그 값에 경우의 수를 +1 더 추가해준다