위장 (해시)

programmers.co.kr/learn/courses/30/lessons/42578?language=python3

코딩테스트 연습 - 위장

def solution(clothes):
    # D: {'headgear': ['yellow_hat', 'green_turban'], 'eyewear': ['blue_sunglasses']}
    D = {}
    for name, kind in clothes:
        if kind not in D:
            D[kind] = [name]
        else:
            D[kind].append(name)

    # cnt: 3 * 2 - 1
    cnt = 1
    for k in D:
        cnt *= len(D[k]) + 1

    return cnt - 1


res = solution([["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]])
print(res)