[프로그래머스] 위장 (Java)

프로그래머스

 

내용만 보면 텍스트를 이리저리 다뤄야할 것 같았는데 막상 읽어보니 경우의 수를 구하는 문제였다. 

 

- HashMap을 선언하여 얼굴, 상의, 하의, 겉옷의 개수를 카운트함

- 종류별 옷의 개수에 안 입는 경우를 더한 후 곱하기 : (얼굴 + 1) * ... * (겉옷 + 1)

- 마지막에 전부 안 입는 경우 1을 빼줌

 

Note

- 고등학교 때 배운 조합식(nCr)을 바로 떠올리긴 했지만 구체적인 수식이 정확하게 기억나지 않아 개념을 찾다보니 결국 전체 경우의 수를 구하면 되는 문제라고 판단하게 되었다.

- 전체 경우의 수를 구하기 위해서는 각 종류별로 안 입는 경우를 고려해야한다는 기초적인 부분을 처음에 놓쳤다는게 아쉬웠다.

 

import java.util.*;

class Solution {
    public int solution(String[][] clothes) {
   
        HashMap<String, Integer> map = new HashMap<>();
        
        for(String[] arr : clothes){
            map.put(arr[1], map.getOrDefault(arr[1], 0) + 1);
        }
        
        int mul = 1;
        
        for (Map.Entry<String, Integer> entry : map.entrySet()) {    
            mul *= (entry.getValue() + 1);        
        }
        
        return mul - 1;
    }
}