[의상/LV2/Python3, C++]

의상 LV2

 

문제 설명
코니는 매일 다른 옷을 조합하여 입는 것을 좋아합니다. 코니는 각 종류별로 최대 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

 

 

알고리즘 고득점 Kit의 해시 문제 중 하나이다.

 

옷의 종류가 A, B, C 3가지가 있다고 가정하자.

A 종류는 2개, B 종류는 3개, C 종류는 4개의 옷이 있다고 하자.

이때의 서로 다른 옷의 조합의 수는 (2+1)(3+1)(4+1) - 1 == 59가지이다.

 

해당 종류를 입지 않는 것도 하나의 경우의 수이므로, 해당 경우를 포함해서 계산해야 한다.

단, 모든 종류에 대해 입지 않는 경우인 1가지를 빼야 한다.

코니는 최소 1가지의 옷은 입기 때문이다.

 

각 종류의 옷의 수를 구한 후, 만든 공식을 사용하여 풀이하였다.

 

 

C++의 unordered_map<string, int>를 통해 풀이하였다.

#include <string>
#include <vector>
#include <unordered_map>

using namespace std;

int solution(vector<vector<string>> clothes)
{
    unordered_map<string, int> hs;
    
    for(auto cloth : clothes)
    {
    // 처음 본 종류면 등록, 이미 본 종류면 1 더하기
        string type = cloth[1];
        if (hs.find(type) == hs.end())
            hs[type] = 1;
        else
            hs[type] += 1;
    }
    
    // 공식을 통해 경우의 수 구하기
    int combi = 1;
    for(auto data : hs)
        combi *=data.second+1;
    
    // 모든 옷을 입지 않은 경우인 1 빼기
    return combi-1;
}

 

 

Python3의 Dictionary를 사용하여 풀이하였다

def solution(clothes):
    dic = {}
    for cloth_name, cloth_type in clothes:
        dic[cloth_type] = dic[cloth_type]+1 if cloth_type in dic else 1
    
    time = 1
    for v in list(dic.values()):
        time *= (v + 1)
    return time - 1

 

 

---