[Baekjoon] 9375: 패션왕 신해빈 - (C3, S3) std::unordered_map

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

입력
첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.
  각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
  다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.
모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

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

9375번: 패션왕 신해빈

 

의상의 종류를 $x$, 각 종류별 의상의 수를 $x[n]$라 하면, 의상을 입을 수 있는 경우의 수는 $\prod (x[n] + 1) - 1$이다. 이때, +1은 해당 종류의 의상을 입지 않는 경우이며, -1은 아무것도 입지 않는 경우이다. 

 

이 문제에서 중요한 것은 의상의 종류이므로, 의상의 종류별 개수를 map으로 정의한다. 이때 정렬은 필요 없으므로 unordered_map을 사용하였다. 

 

의상의 이름과 종류를 입력받을 때, 그 종류가 이미 map에 존재한다면 해당 종류의 의상 수를 1 증가시키고, 존재하지 않으면 새로 추가한다. 

 

모든 입력이 종료되면 map의 키(종류) 별 값(개수) + 1을 곱하여 누적하고, 최종 값에서 1을 빼 출력한다. 

 

#include <iostream>
#include <unordered_map>
using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T, N;
    cin >> T;

    while (T--)
    {
        cin >> N;
        string name, type;
        int days = 1;

        if (N == 0) // 의상이 0이면 0 출력
        {
            cout << 0 << '\n'; 
            continue;
        }

        unordered_map<string, int> clothes; // 의상의 종류별 개수
        for (int i = 0; i < N; i++)
        {
            cin >> name >> type;
            if (clothes.count(type)) // 종류가 존재하면 개수 + 1
                clothes[type]++;
            else
                clothes[type] = 1; // 없으면 추가
        }

        for (const auto &[type, count] : clothes) // 모든 경우의 수
            days *= (count + 1);

        days--; // 안 입는 경우 제거거
        cout << days << '\n';
    }

    return 0;
}

count는 해당 키가 존재하면 1을, 존재하지 않으면 0을 반환한다.