위장 (Java)

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

코딩테스트 연습 - 위장

 

프로그래머스 코딩테스트 고득점 Kit 위장 문제 풀이이다.

종류 별로 하나씩만 입을 수 있으므로 맵에 종류를 키로, 종류별 의류 수를 값으로 넣어준다.

예를 들어 종류별 의류 수가 2개면 0(안 입기), 1(1번 입기), 2(2번 입기)의 세 가지 경우가 있을 수 있고, 그 경우들을 모두 곱해주면 경우의 수를 구할 수 있다. 아무 것도 안 입는 경우는 없으므로 모두 0인 경우를 하나 빼주었다.

 

import java.util.*;

class Solution {
    public int solution(String[][] clothes) {
        int answer = 1;
        
        // clothes 배열의 종류를 key로, 종류별 개수를 value로 map에 넣어줌 
        HashMap<String, Integer> map = new HashMap<>();
        for (int i = 0; i < clothes.length; i++) {
            // 종류, 개수의 쌍으로 넣어줌
            // 개수는 map에 그 종류가 없으면 0을 넣어줌
            map.put(clothes[i][1], map.getOrDefault(clothes[i][1], 0) + 1);
        }
        
        // 종류, 개수의 쌍으로 이루어진 map에서 key만 추출
        Set<String> set = map.keySet();
        for (String key : set) {
            // answer에 value + 1 한 값을 누적하여 곱해줌
            answer = answer * (map.get(key) + 1);
        }
        
        // 안 입는 경우
        return answer - 1;
    }
}