[BackJoon]9375번

문제

 

해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 

예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 

해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?

 

입력

 

첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.

 

• 각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
• 다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.

모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

 

출력

 

각 테스트 케이스에 대해 해빈이가 알몸이 아닌 상태로 의상을 입을 수 있는 경우를 출력하시오.

 

풀이

 

처음 문제를 읽고선 아주 당연하게도 멱집합 문제라고 생각했다.

 

그래서 여러 번에 걸쳐서 다양하게 멱집합을 구성해 봤지만 메모리 초과로 실패.

 

동적계획법으로 풀어보려고 했지만 역시 실패했다.

 

멱집합을 구현해서 풀어보고 싶었는데..

 

어쨌거나 통과는 해야하니까 조금 단순한 계산으로 돌아와서 풀어보았다.

 

먼저 예제로 준 문제를 보면 아래와 같다.

2
3
hat headgear
sunglasses eyewear
turban headgear
3
mask face
sunglasses face
makeup face

첫 번째 예시를 보면 헤드기어 두 개, 아이웨어 하나이다.

 

이때 의상을 종류별로 나눠보면 아래와 같다.

hat headgear
turban headgear

sunglasses eyewear

그럼 이때 입을 수 있는 총 조합의 경우는 몇 개일까?

 

헤드기어 두개 중 하나를 고르는 경우 X 아이웨어 중 하나를 고르는 경우라고 생각해서 2개라고 생각할 수도 있다.

 

하지만 문제의 조건에 보면 하나라도 의상을 걸치고 있으면 알몸으로 치지 않는다.

 

그러니까 사실은 (2+1)개의 의상에서 하나를 고르는 경우의 수 X (1+1)개의 의상에서 하나를 고르는 경우의 수에서

 

아무것도 입지 않은 경우를 빼주는 문제가 된다. 답은 5.

 

하나만 고르는 문제이기 때문에 조합까지 갈 것도 없이 동일한 파트의 의상이 몇 개만 있는지 체크하면 된다.

 

해시맵을 구현해서 아래와 같이 풀었다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;
import java.util.StringTokenizer;

public class Prob9375_2 {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int t = Integer.parseInt(br.readLine());

        for (int i = 0; i < t; i++) {

            int n = Integer.parseInt(br.readLine());
            Map<String, Integer> cntMap = new HashMap<>();

            for (int j = 0; j < n; j++) {

                StringTokenizer st = new StringTokenizer(br.readLine());

                String value = st.nextToken();
                String key = st.nextToken();

                if (cntMap.containsKey(key)) {
                    cntMap.replace(key, cntMap.get(key) + 1);
                } else {
                    cntMap.put(key, 1);
                }
            }

            int result = 1;

            for (int num : cntMap.values()) {
                result *= (num + 1);
            }

            System.out.println(result - 1);
        }
    }
}