해시 함수 - 프로그래머스 의상

 

<문제>

 

코딩테스트 연습 - 의상 | Programmers School

 

서로 다른 옷을 입는 가능한 조합을 구하는 함수. 

같은 종류의 옷 중에서 0개 또는 1개 선택할 수 있고, 전체 옷 중에서는 최소한 1개를 선택해야 한다. 

 

 

<내 풀이>

from collections import Counter #배열에 특정 원소가 몇 개 있는지 계산해주는 클래스

#입력 배열 예시: [["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]

def solution(clothes):
    arr=[]   # 옷 종류를 담을 배열
    dict={}  #{옷 이름: 옷 종류}를 받는 딕셔너리
    for item in clothes:
        dict[item[0]]= item[1]  #옷 이름을 key로, 옷 종류를 value로 딕셔너리에 추가한다.
        
    for value in dict.values():
        arr.append(value) #딕셔너리에 있는 아이템들의 옷 종류를 배열에 중복 허용해서 담는다.
    counter= Counter(arr) #counter이용해서 딕셔너리 value값에 옷 종류가 몇번씩 나왔는지 저장한다.

    res=1

    for value in counter.values():
        res *= (value+1) #옷 종류 중 한 개 또는 0개를 선택하는 경우의 수= value+1개.를 모두 곱한다.
    res-=1 #res에서 모든 옷 종류에서 한 개도 선택하지 않은 경우를 뺀다.

    return res

 

collections 모듈의 Counter 클래스를 처음 써 보았다.