[프로그래머스 - 해시] Lv2.위장

종류별 가짓수를 체크하기 위해 Map을 사용한다. 

Map의 키가 공통으로 묶을수 있는 종류가 된다.

import java.util.*;
class Solution {
    public int solution(String[][] clothes) {
        Map<String, Integer> counts = new HashMap<String, Integer>();
        
        for(String[] s : clothes) {
            String type = s[1];
            counts.put(type, counts.getOrDefault(type, 0) + 1);
        }
        
        int answer = 1;
        for(Integer num : counts.values()) {
            answer *= num + 1;//사용하지 않는 경우 추가하여 곱한다.
        }
        return answer-1 ;//모두다 사용하지 않는 경우를 뺀다.
    }
}

새롭게 알게된것

Map.getOrDefault(key값, default value)

map의 값을 꺼냈을 때 null인 경우 삼항 연산자를 통해 길어졌지만 getOrDefault메서드를 통해 코드를 줄일 수 있다.

 

stream을 사용한 리팩터링

import java.util.*;
class Solution {
    public int solution(String[][] clothes) {
       int answer = Arrays.stream(clothes)
           .map(c ->c[1])
           .distinct()
           .map(type -> (int) Arrays.stream(clothes).filter(c -> c[1].equals(type)).count())
           .map(c -> c +1)
           .reduce(1,(c,n) -> c * n);
           return answer -1;
    }
}

프로그래머스 java 커뮤니링 강의를 듣고 정리한 내용입니다.