[프로그래머스] L2 위장

문제

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

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

종류	이름
얼굴	동그란 안경, 검정 선글라스
상의	파란색 티셔츠
하의	청바지
겉옷	긴 코트

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

제한사항

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

풀이

내용을 간략화 해보면 결국 경우의 수 문제입니다. 각 종류의 수에 따른 조합을 맞춰입는 경우의 수를 계산하는 문제라고 볼 수 있습니다.

따라서 위의 예제를 기준으로 공식을 새워보면 다음과 같다고 할 수 있습니다.

얼굴 2개, 상의 1개, 하의 1개, 겉옷 1개입니다. 각 부위에 입지 않는 조합이 가능하기 때문에 각 종류 수에  +1(입지 않음) 을 해준 후 계산을 해야합니다.

얼굴 (2 + 1) * 상의 (1 + 1) * 하의 (1 + 1) * 겉옷 (1 + 1) = 3 * 2 * 2 * 2 = 24

하지만 여기서 제한사항으로 나와있는 것이 최소 1개의 의상은 입는다 입니다. 따라서 모두 입지 않는 경우의 수를 하나 제외해주는 예외처리가 필요합니다.

따라서 계산된 결과에 -1를 해주어 24 -1 = 23이 결과로 나오겠습니다.

public int solution(String[][] clothes) {
    HashMap<String, Integer> clothMap = new HashMap<>();
    int value;
    for (String[] clothSet: clothes) {
        value = 1;
        if (clothMap.containsKey(clothSet[1])) {
            value = clothMap.get(clothSet[1]) + 1;
        }
        clothMap.put(clothSet[1], value);
    }


    int total = 1;
    for (int count: clothMap.values()) {
        total *= (count + 1);
    }
    return total - 1;
}

제 경우엔 각 부위의 종류를 계산하기 위해 HashMap을 이용하여 계산을 하였습니다. 제한사항에 같은 이름의 의상이 존재하지 않는다고 했기 때문에 해당 부분에 대한 처리는 필요하지 않습니다. 따라서 어떤 부위에 값이 들어왔다면 해당 부위의 수를 늘려주도록 처리하였습니다.

결과