[ 프로그래머스/JAVA] 위장 SPY 해시

1. 문제

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

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

 

종류이름

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

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

 

 

제한사항

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

 

입출력 예

[["yellowhat", "headgear"], ["bluesunglasses", "eyewear"], ["green_turban", "headgear"]]	5
[["crowmask", "face"], ["bluesunglasses", "face"], ["smoky_makeup", "face"]]	3

 

 

입출력 예 설명

 

 

예제 #1
headgear에 해당하는 의상이 yellow_hat, green_turban이고 eyewear에 해당하는 의상이 blue_sunglasses이므로 아래와 같이 5개의 조합이 가능합니다.

1. yellow_hat

2. blue_sunglasses

3. green_turban

4. yellow_hat + blue_sunglasses

5. green_turban + blue_sunglasses

 

예제 #2
face에 해당하는 의상이 crow_mask, blue_sunglasses, smoky_makeup이므로 아래와 같이 3개의 조합이 가능합니다.

1. crow_mask

2. blue_sunglasses

3. smoky_makeup

2. 해결방법(코드)

import java.util.*;
class Solution {
    public int solution(String[][] clothes) { 
        //경우의 수 구하는 문제(return 수)
        int answer = 1;
        Map<String, Integer> map = new HashMap<>();
        
        //의상 종류(Key), 해당 의상의 개수(Value)
        for(String[] clothes_arr : clothes){
            String key = clothes_arr[1];
            map.put(key, map.getOrDefault(key, 0) + 1);
        }
        
        for(String key : map.keySet()){
            //안 입는 경우 더하기
            answer *= map.get(key) + 1;
        }
        
        //최소 한개의 의상은 입는다고 해서 아예 안입는 경우 제외
        return answer - 1;
    }
}

 

문제를 읽고 경우의 수 구하는 풀이로 결정을 했다.

여러 종류 의상을 동시에 착용이 가능해서 곱하기(*)를 사용한다. (동시에 불가능 하면 더하기(+) 이용!)

 

한 의상을 입을 수 있는 경우의 수 = 한 의상의 개수 + 1(안 입을 경우)

 

문제에서 아예 안 입을 경우를 고려하는지 확인 후 -1을 추가로 해준다.

 

 

 

예) 상의 2개, 하의 4개

상의를 입을 수 있는 경우의 수 = 2개 + 1 = 3

하의를 입을 수 있는 경우의 수 = 4개 + 1 = 5

 

동시에 착용이 가능 ==> 3 * 5 = 15개

 

문제에서 최소 한 개의 의상을 입는다는 말이 있으므로 

 

정답 : 15 - 1 = 14개 

 

 

 

3. 오답

import java.util.*;
class Solution {
    public int solution(String[][] clothes) { 
        //경우의 수 구하는 문제(return 수)
        int answer = 1;
        Map<String, Integer> map = new HashMap<>();
        
        for(String[] clothes_arr : clothes){
            String key = clothes_arr[1];
            //빈도수 출력     
            Integer count = map.get(key);
            
            // 이 부분을 .getOrDefault 사용함
            if(count == null){
                map.put(key, 1);
            }else{
                map.put(key, count+1);
            }
        }
        
        //안 입는 경우를 고려하지 못함
        for(String key : map.keySet()){
            answer = answer * map.get(key);
        }
        
        if(map.size() < 2){
            answer = clothes.length;
        }else{
            answer = answer + clothes.length;
        }
        
        
        return answer;
    }
}

• 경우의 수 문제를 파악했지만, 안 입는 경우를 고려하지 못했다.
• if문 코드를 Map의 getOrDefault 메소드를 이용해 개선을 했다.