프로그래머스/Lv. 2

[프로그래머스 코딩테스트] 의상(Java)

Sigfriede 2023. 7. 3. 01:00

  문제 설명

  코니는 매일 다른 옷을 조합하여 입는 것을 좋아합니다.

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

종류 이름
얼굴 동그란 안경, 검정 선글라스
상의 파란색 티셔츠
하의 청바지
겉옷 긴 코트
  • 코니는 각 종류별로 최대 1가지 의상만 착용할 수 있습니다. 예를 들어 위 예시의 경우 동그란 안경과 검정 선글라스를 동시에 착용할 수는 없습니다.
  • 착용한 의상의 일부가 겹치더라도, 다른 의상이 겹치지 않거나, 혹은 의상을 추가로 더 착용한 경우에는 서로 다른 방법으로 옷을 착용한 것으로 계산합니다.
  • 코니는 하루에 최소 한 개의 의상은 입습니다.

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

 

  제한사항

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

 

  입출력 예

clothes return
[["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]] 5
[["crow_mask", "face"], ["blue_sunglasses", "face"], ["smoky_makeup", "face"]] 3
import java.util.HashMap;
import java.util.Collection;
class Solution {
    public int solution(String[][] clothes) {
        HashMap<String, Integer> map = new HashMap<>();
        for (int i = 0; i < clothes.length; i++) {
            map.put(clothes[i][1], map.getOrDefault(clothes[i][1], 0) + 1);
        }
        Collection<Integer> values = map.values();
        int mul = 1;
        for (int value: values) {
            mul *= (value + 1);
        }
        int answer = mul - 1;
        return answer;
    }
}

  제네릭 타입으로 String과 Integer를 갖는 HashMap을 생성합니다.

  for문이 clothes의 길이만큼 순회합니다.

  put 메소드로 map에 의상의 종류와 개수를 추가합니다. getOrDefault 메소드를 이용하여 map에 없다면 추가하고, 있다면 value에 1을 더해줄 것입니다.

  제네릭 타입으로 Integer를 갖는 Collection을 생성합니다. values 메소드를 이용하여 map의 value를 전부 담습니다.

  변수 mul을 생성합니다.

  for-each문에서 values를 value에 하나씩 할당합니다.

  mul에 value + 1을 곱하기 할당 해나갑니다. 이 문제에서는 특정한 의상을 입거나 입지 않는 상황으로 나뉩니다. 따라서 각 의상에 1씩 더해주어 이 상황을 만들어주는 것입니다. 그러나 전부 입지 않는 경우의 수도 있을 것이므로 구한 값에서 이 한 가지의 경우를 빼주는 것입니다.

  따라서 mul에서 1을 뺀 값을 answer에 할당합니다.