[Programmers] 의상 - Java

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

프로그래머스

---

1. 문제

[의상종류, 카테고리]가 담긴 2차원 배열 clothes가 주어진다.

서로 다른 옷을 조합해서 입을 수 있는 수를 반환한다.

주의할 점은

1. 옷을 하나만 입을 수도 있다.

2. 옷을 아무것도 입지 않을 수는 없다.

 

2. 풀이

해시 + 수학으로 쉽게 풀 수 있다.

총 조합 수 = (종류별 의상 수+1) * (종류별 의상 수+1)  * ... * (종류별 의상 수+1)  - 1

을 해주면 총 조합의 수가 된다. 

+1을 해주는 이유는 그 옷을 입지 않았을 경우의며,

-1을 해주는 이유는 옷을 아무것도 입지 않았을 경우이다.

 

예를 들면

[["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]의 경우,

headgear : 2

eyewear : 1

인 상태이다.

(2+1) * (1+1) - 1 = 5, 답은 5가 된다.

[["crow_mask", "face"], ["blue_sunglasses", "face"], ["smoky_makeup", "face"]]

face : 3

(3+1) - 1 = 3, 답은 3이 된다.

 

3. 코드

import java.io.*;
import java.util.*;

class Solution {
    public int solution(String[][] clothes) {
        
        Map<String, Integer> map = new HashMap<> ();
        for(String[] arr : clothes) {
            if (map.containsKey(arr[1])) {
                map.put(arr[1], map.get(arr[1])+1);
            }
            else {
                map.put(arr[1], 1);
            }
        }
        
        int answer = 1;
        for (String key : map.keySet()) {
            answer *= (map.get(key) + 1);
        }
        answer--;
        
        return answer;
    }
}

4. 사담

감을 잡자 감을 잡자

할 수 있다 할 수 있다 ~