[Java] 프로그래머스 Lv.2 : 위장

https://programmers.co.kr/learn/courses/30/lessons/42578?language=java 

코딩테스트 연습 - 위장

 

풀이

경우의 수 계산법만 알면 간단히 풀 수 있다.

예를 들어, 상의 A개, 하의 B개가 주어진다면 서로 다른 옷의 조합의 수는 다음과 같이 계산할 수 있다.

1. 상의든 하의든 1개만 선택해서 입는 경우 = A+B
2. 상의1개 하의 1개 선택해서 입는 경우 = A*B
3. 아무것도 입지 않는 수 = 1

최종 공식 = A+B+(A*B)+1 = (A+1)*(B+1)

하지만 우리는 아무것도 입지 않는 수는 빼줘야하기 때문에 최종적으로 -1을 해주도록 한다.

HashMap에 key값으로는 의상의 종류, value로는 그 개수를 넣어줄 것인데, 의상의 종류에는 중복이 있으므로 전에 완주하지 못한 선수 문제를 풀 떄 공부했던 getOrDefault() 메서드를 사용해 중복처리를 해준다.

import java.util.*;
class Solution {
    public int solution(String[][] clothes) {
        int answer = 1;
        HashMap<String, Integer> hm = new HashMap<>();
        for(int i=0; i<clothes.length; i++){
            hm.put(clothes[i][1], hm.getOrDefault(clothes[i][1],0)+1);
        }
        for(Map.Entry<String, Integer> entry : hm.entrySet()){
            answer *= (entry.getValue()+1);
        }
        return answer-1;
    }
}