[PS] 99클럽 코테 스터디 6일차 TIL (의상)

문제

문제 링크

풀이

class Solution {
  public int solution(String[][] clothes) {
    int answer = 1;
    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);
    }

    for (int count : map.values()) {
      answer *= count + 1;
    }

    return answer - 1;
  }
}

문제에서 처럼 종류 별로 몇 개씩 있는지 저장할 HashMap 을 만든다.

종류	갯수
얼굴	2
상의	1
하의	1
겉옷	1

map.put(clothes[i][1], map.getOrDefault(clothes[i][1], 0) + 1); 이렇게 하면 key 에 옷의 종류가 들어가고, value 로 갯수를 넣을 수 있다. getOrDefault() 메소드는 키에 대한 값이 있으면 그 값을 반환하고 없으면 0을 반환한다. 계속 1씩 늘려가야 하므로 +1 을 해준다.

만약 A 종류 의상이 3개, B 종류 의상이 2개인 경우를 생각해보자.

• A 또는 B 만 입는 경우 : 3 + 2
• A 와 B 를 섞어 입는 경우 : 3 * 2
• 총 조합 갯수 : 3 + 2 + (3 * 2) = 11 개

A 종류의 의상이 a1, a2, a3 이 있고, B 종류의 의상이 b1, b2 일때 선택 방법의 수는 의상은 다음과 같이 선택 할 수 있다.

• A 종류 선택 방법 : 선택안함 / a1 / a2 / a3 (총 4 가지)
• B 종류 선택 방법 : 선택안함 / b1 / b2 (총 3 가지)

그러면 A 와 B 의 총 조합 수는 (A 종류 선택 방법) * (B 종류 선택 방법) = 12 개가 나온다. 여기서 둘다 선택 안하는 경우의 수 1 을 빼주면 11 이 나온다.

총 의상 선택 경우의 수 : (A 종류 + 1) * (B 종류 + 1) - 1

이것을 코드로 구현하면 다음과 같다.

// (A + 1)(B + 1)...
int answer = 1;
for (int count : map.values()) {
  answer *= count + 1;
}
return answer - 1;