[Baekjoon/JAVA] 9375번 - 패션왕 신해빈

[백준] Baekjoon Online Judge

문제로 이동

 

문제

해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?

 

입력

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

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

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

 

출력

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

 

---

예제 입력	예제 출력
2 3 hat headgear sunglasses eyewear turban headgear 3 mask face sunglasses face makeup face	5 3

---

풀이

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

public class Main {
    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++) {
            Map<String, Integer> clothes = new HashMap<String, Integer>();
            int n = Integer.parseInt(br.readLine()); // 해빈이가 가진 의상의 수

            // 해빈이가 가진 의상을 종류별로 몇 개 있는지 map에 저장
            for (int j = 0; j < n; j++) {
                String[] s  = br.readLine().split(" ");

                if(!clothes.containsKey(s[1]))
                    clothes.put(s[1], 1);
                else
                    clothes.put(s[1], clothes.get(s[1]) + 1);
            }

            int cnt = 1; // 의상을 입을 수 있는 경우의 수 구하기
            Set<String> keySet = clothes.keySet();

            for (String key : keySet) cnt *= (clothes.get(key) + 1);

            // 출력
            System.out.println(cnt - 1);
        }
    }
}

 

            Map<String, Integer> clothes = new HashMap<String, Integer>();
            int n = Integer.parseInt(br.readLine()); // 해빈이가 가진 의상의 수

            // 해빈이가 가진 의상을 종류별로 몇 개 있는지 map에 저장
            for (int j = 0; j < n; j++) {
                String[] s  = br.readLine().split(" ");

                if(!clothes.containsKey(s[1]))
                    clothes.put(s[1], 1);
                else
                    clothes.put(s[1], clothes.get(s[1]) + 1);
            }

의상의 이름과 종류를 입력 받는 부분이다. 의상의 이름은 어차피 서로 모두 다르고, 출력에 사용하지 않기에 무시하고 의상의 종류만 사용한다. map에 입력 받은 의상의 종류가 key로 존재하지 않는다면 map에 그 값을 key로 한 칸을 주고, 기존에 key로 존재한다면 value를 1 증가시킨다. 즉, map의 value는 해당 의상의 종류를 몇 개 입력 받았는지를 의미한다.

 

            int cnt = 1; // 의상을 입을 수 있는 경우의 수 구하기
            Set<String> keySet = clothes.keySet();

            for (String key : keySet) cnt *= (clothes.get(key) + 1);

            // 출력
            System.out.println(cnt - 1);

예를 들어 머리에 쓰는 의상이 3개 있다면 머리에 쓰는 의상을 입는 경우의 수는 아예 안 입는 것을 포함해서 4가지이다. 그러므로 map의 value들을 1씩 증가시켜서 cnt에 곱해주었다. 그리고 이렇게 되면 모든 옷을 아예 안 입는 경우가 생기므로 1을 빼서 출력했다.