본문 바로가기
JAVA/알고리즘

[ 해시 Level2 ] 프로그래머스 위장 - 자바 JAVA

by hak0205 2021. 7. 18.
반응형

 

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

 

코딩테스트 연습 - 위장

 

programmers.co.kr

 

- 문제 설명

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

 

종류 이름
얼굴 동그란 안경, 검정 선글라스
상의 파란색 티셔츠
하의 청바지
겉옷 긴 코트

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해 주세요.

제한사항

  • clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
  • 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
  • 같은 이름을 가진 의상은 존재하지 않습니다.
  • clothes의 모든 원소는 문자열로 이루어져 있습니다.
  • 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
  • 스파이는 하루에 최소 한 개의 의상은 입습니다.

- 입출력 예

clothes return
[["yellowhat", "headgear"], ["bluesunglasses", "eyewear"], ["green_turban", "headgear"]] 5
[["crowmask", "face"], ["bluesunglasses", "face"], ["smoky_makeup", "face"]] 3

 

- 입출력 예 설명

예제 #1
headgear에 해당하는 의상이 yellow_hat, green_turban이고 eyewear에 해당하는 의상이 blue_sunglasses이므로 아래와 같이 5개의 조합이 가능합니다.

1. yellow_hat                       

2. blue_sunglasses                       

3. green_turban

4. yellow_hat + blue_sunglasses                                             

5. green_turban + blue_sunglasses

 

예제 #2
face에 해당하는 의상이 crow_mask, blue_sunglasses, smoky_makeup이므로 아래와 같이 3개의 조합이 가능합니다.

1. crow_mask                     

2. blue_sunglasses                       

3. smoky_makeup


프로그래머스 위장 자바 코드

import java.util.HashMap;

class Solution {
    public int solution(String[][] clothes) {
        HashMap<String, Integer> hm = new HashMap();
        int answer = 1;

        //의상종류로 입을 수 있는 수를 세어줍니다.
        for (int i = 0; i < clothes.length; i++) {
            hm.put(clothes[i][1], hm.getOrDefault(clothes[i][1], 0)+1 );
        }
        //getOrDefault(key, defaultValue)
        //key : 값을 가져와야 하는 요소의 키입니다.
        //defaultValue : key에 맵핑되어있는 값이 존재한다면 해당 값을 가져오고 없다면 
        //               defaultValue값을 반환합니다.

        for (int i : hm.values()) {
            answer *= (i+1); //해당 종류 단독으로 착용했을 경우가 있으므로 +1을 해줍니다.
        }

        //아무것도 입지 않았을경우는 구하지 않으므로 -1을 해줍니다.
        return answer-1; 
 }
}

정리

의상 종류를 중심으로 의상 종류가 겹치지 않게 입을 수 있는 경우의 수를 구하라는 것입니다.

즉, 조합의 원리만 알고 있다면 풀 수 있는 문제입니다.

 

  1. 의상종류로 옷을 조합할 수 있는 수를 세어줍니다.
  2. 그리고 해당 종류 단독으로만 입었을 경우도 있으므로 경우의 수를 구할 때 +1을 해줍니다. 예를 들어 의상 종류가 headgear라면 headgear 단독으로 착용했을 경우가 있습니다.
  3. 3. 구한 경우의 수에서 아무것도 입지 않는 경우 1가지를 빼줍니다.

코드의 주석도 참고하시면 이해하시기 편합니다.

 

알고리즘을 이해하고 풀이를 하신다면 기억에 오래 남습니다. 

 

감사합니다.

 
반응형

댓글