[프로그래머스] 위장 - JAVA

[문제 링크]

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

import java.util.*;
class Solution {
    public int solution(String[][] clothes) {
        HashMap<String, Integer> hashMap = new HashMap<>();

        for (String[] cloth : clothes) {
            //만약 해시에 키가 없으면 1개로 카운트 해서 추가
            if (!hashMap.containsKey(cloth[1])) {
                hashMap.put(cloth[1], 1);

            //해시에 키가 존재하면 이전 카운트에 1 추가
            } else if (hashMap.containsKey(cloth[1])) {
                hashMap.put(cloth[1], hashMap.get(cloth[1]) + 1);

            }
        }

        //경우의 수를 위해 answer에 곱해줄 것이니까 1로 해줌
        int answer = 1;

        //해시의 밸류에 +1(옷을 안입는 경우의 수를 추가해줌)을 해주고 곱해준다(전체 경우의 수 구하기)
        for (Integer value : hashMap.values()) {
            answer *= value + 1;
        }

        //옷 전체를 안입는 경우는 없기 때문에 해당 경우의 수를 빼준다.

        return answer-1;
    }
}

문제를 보고 키 밸류 쌍으로 하면 편하겠다고 생각함.

 

처음에는 문제의 조건 그대로 [["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]이런 식으로 넣었음.

이렇게 넣고 밸류의 갯수를 카운팅하려고 했는데, 문제가 잘 안풀릴 것같았음.

예시는 모자, 안경인데 제한사항에서 의상 수가 1개 이상 30개 이하라고 하니 복잡하고 생각도 안남.

 

그래서 접근 방식을 바꿈.

보니까 yellow_hat, blue_sunglasses이런 애들은 필요가 없음. 

결국 headgear같은 종류가 하나씩밖에 못 옷다는 게 중요했음.

그러면 이 다차원 배열에서 옷의 종류가 몇 개가 있는지 세고, 조합의 갯수를 세기 위해서 나온 옷의 종류를 곱해주면 되겠다고 생각함.

 

그래서 키를 옷의 종류, 밸류를 종류의 갯수로 설정함.

 

그렇게 곱하려고 했는데, 생각해보니 옷을 안입는 경우도 있음.

그래서  계산을 할때 종류의 갯수에 +1을 해줌. (경우의 수에 옷을 안입는 경우가 추가된 것)

 

답이 안나옴

 

이렇게 하면 옷을 전부 안입는 경우가 생김.

문제의 조건 중 스파이는 최소 한 개 이상의 옷을 입는다고 되어있음.

 

그래서 최종적으로 나온 answer에 -1 (옷을 전부 안입는 경우)를 해줌.