[프로그래머스] 위장

문제 보기

이 문제는 해쉬 문제이다. (그런데 정작 해쉬는 잘 안 쓰인 것 같다..)

처음엔 간단해 보였는데 경우의 수를 잘 못해서 한참 헤맸다..

문제를 푼 로직은 다음과 같다.

 

1. 먼저 옷의 각 종류에 대해서 옷의 개수를 센다. (옷의 이름은 중요하지 않다. 개수만 알면 된다.) (7~11번째 줄)
2. 만약 옷의 종류가 한 가지라면 옷의 개수만큼 반환한다. (스파이는 최소 1개 이상의 옷을 입어야하기 때문) (13~16번째 줄)
3. 만약 옷의 종류가 한 가지 이상이라면, (종류 1의 개수 + 1) * (종류 2의 개수 + 1) * ...  - 1을 반환한다.
   (옷의 종류에서 1을 더하는 이유는 해당 종류의 옷을 안 입을 수 있기 때문이고, 마지막에 - 1을 하는 이유는 스파이는 최소 1개 이상의 옷을 입어야 하므로, 아무것도 안 입은 경우의 수를 빼야 하기 때문이다.)
   (17~23번째 줄)

 

from collections import Counter
 
 
def solution(clothes):
    answer = 0
    
    # make categories dictionary
    categories = list()
    for name, category in clothes:
        categories.append(category)
    categories = Counter(categories)
    
    length = len(categories.values())
    # if there's only one category
    if length == 1:
        answer = list(categories.values())[0]
    # if there are one more category
    elif length > 1:
        temp = 1
        for value in list(categories.values()):
            # get the combination including the case of NOT wearing an item
            temp *= (value + 1)
        answer += temp - 1
    
    return answer