[알고리즘] 프로그래머스 위장 -해시(hash)-

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

코딩테스트 연습 - 위장

 

프로그래머스 LEVEL2 의 해시 유형 문제를 풀어봤습니다. ㅎㅎ

처음에 해시로 접근할 때 value 값을 List로 할려했는데 생각해보니 같은 옷이 나오는 경우는 생각을 안해도 되서 key 값은 옷의 종류, value 값은 해당 옷 종류의 옷 개수로 설정해서 쉽게 풀 수 있었습니다.

 

hash 문제를 풀때 getOrDefault() 함수를 참 많이 사용하는 거 같습니다 저같은 경우는.. 

 

그리고 이 문제에서는 확률과 통계 지식도 필요했는데요.

다음과 같이 이 문제 수학지식에 대해 잘 설명해논 글이 있어 가져와봤습니다. !!

 

출처: https://2ssue.github.io/algorithm/programmers_42578/

 

 

 

풀이는 다음과 같습니다.

 

[Java]

import java.util.HashMap;

class Solution {
    public static void main(String[] args) {
        Solution solution = new Solution();
        solution.solution(new String[][]{{"yellow_hat", "headgear"}, {"blue_sunglasses", "eyewear"}, {"green_turban", "headgear"}});
    }

    public int solution(String[][] clothes) {
        int answer = 1;
        HashMap<String, Integer> map = new HashMap<>(); // (의상의종류 , 옷개수)
        for (int i = 0; i < clothes.length; i++) {
            if (map.getOrDefault(clothes[i][1], 0) != 0) {
                map.put(clothes[i][1], map.get(clothes[i][1]) + 1);
            } else {
                map.put(clothes[i][1], 1);
            }
        }
        for (String key : map.keySet()) {
            answer *= map.get(key) + 1;
        }
        answer -= 1;
        return answer;
    }
}

 

 

[Kotlin]

import java.util.HashMap

internal class Solution {
    fun solution(clothes: Array<Array<String>>): Int {
        var answer = 1
        val map : HashMap<String,Int> = HashMap<String, Int>() // (의상의종류 , 옷개수)
        for (i in clothes.indices) {
           clothes[i][1].let {
                clothType ->
                if (map.getOrDefault(clothType, 0) != 0) {
                    map[clothType] = map[clothType]!! + 1
                } else {
                    map[clothType] = 1
                }
            }
        }
        for (key in map.keys) {
            answer *= map[key]!! + 1
        }
        answer -= 1
        return answer
    }
}

 

 

 

댓글과 공감은 큰 힘이 됩니다. 감사합니다!!