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

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

코딩테스트 연습 - 위장 | 프로그래머스

코드

import java.util.*;

class Solution {
    public int solution(String[][] clothes) {
        HashMap<String, Integer> map = new HashMap<String, Integer>();
        int answer = 1;
        
        for(int i=0; i<clothes.length; i++){
            // 현재 의상의 종류가 존재하지 않으면 1개 넣고, 존재하면 +1
            if(map.get(clothes[i][1]) == null)  map.put(clothes[i][1], 1);
            else    map.put(clothes[i][1], map.get(clothes[i][1]) + 1);
        }
        
        for(String keys : map.keySet())
            answer *= (map.get(keys) + 1);
        
        return answer-1;
    }
}

풀이

자바의 컬렉션 프레임워크중 List에 대해선 어느정도 사용은 해보았지만, Map에 대해선 거의 무지해서 해시에 대해 어느정도 공부를 해야 할 것도 같았고, 최근 알고리즘을 풀면 해시맵을 사용했을때, 사용해야할 때 풀리는 문제들도 어느정도 존재해서 공부해야 할 것 같아서 풀어보았다...(물론 직접 풀진 못했다 .. ㅎㅎ)

 

이 문제에서 필요한 것은 옷의 이름이 아닌 옷 종류의 개수가 몇개 존재하느냐이다.

{옷 종류} : {총 갯수} 와 같이 Key : Value 형태와 비슷한 구조이다.

왜 ? 경우의수를 구하기 위해서다. 간단하게 풀기 위해선 이 경우의 수를 구하는 알고리즘이 전부다..

그럼 경우의 수는 어떻게 구하는지 보자.

입을 수 있는 경우의 수는 각 옷의 종류 +1 을 모두 곱한 뒤 -1 을 해주면 된다.

각 종류의 옷들은 모두 입거나 안입거나 2가지의 경우가 있기 때문에, +1씩 해줘야 한다.

만약 예제에서 처럼 옷의 종류가 "headgear" , "eyewear" 의 2종류가 있고, 의상은 각각 2개, 1개가 있다면

(2+1) * (1+1) - 1 = 5 가 정답이 된다.

 

옷을 하나도 입지 않는 경우도 생기므로 -1을 해줘야 한다.