[JAVA/Map] 프로그래머스 위장

1. 문제

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

코딩테스트 연습 - 위장

 

2. 풀이

map을 사용해서 풀이했다. map<옷 종류, 개수> 형식으로 옷의 종류마다 개수를 세었다.

 

결과를 구할 때는,
headgear: 2개
face: 3개
eyewear: 1개

각 종류의 옷을 선택하는 경우 + 해당 종류의 옷을 아예 안입을 경우를 고려해
옷의 종류 + 1을 모두 곱해준다.

즉, 위의 경우에서는 (2+1) * (3+1) * (1+1)을 한다.

마지막으로 모든 옷을 입지 않을 경우를 제외해서 -1을 해준다.

 

3. 코드

 

import java.util.HashMap;
import java.util.Set;

class Solution {
    public int solution(String[][] clothes) {
        int answer = 1;

        HashMap<String, Integer> map = new HashMap<String, Integer>();

        for(int i = 0; i < clothes.length; i++){
            map.put(clothes[i][1], map.getOrDefault(clothes[i][1], 0) + 1);
        }

        Set<String> keySet = map.keySet();

        for(String s: keySet){
            answer *= map.get(s) + 1;
        }

        return answer - 1;
    }
}