[프로그래머스 코테 연습] 위장

문제 요약

clothes의 각 행은 [의상의 이름, 의상 종류]로 이루어져 있다.
같은 이름을 가진 의상은 존재하지 않고, clothes의 모든 원소는 문자열로 이루어져 있다.
모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있다. 스파이는 하루에 최소 한 개의 의상은 입어야 할 때, 서로 다른 종류 옷의 조합 수를 return 하라.

답안 예시

def solution(clothes):
    answer = {}
    
    for i in clothes:
        if i[1] in answer:
            answer[i[1]] += 1
        else:
            answer[i[1]] = 1
            
    cnt = 1
    
    # (종류 + 1) 곱하고 1 빼면 되는 이유
    # (a + 1)(b + 1)(c + 1) - 1 = (a + b + c) + (ab + bc + ca) + abc
    
    for i in answer.values():
            cnt *= (i+1)
    
    return cnt - 1

느낀점

• 경우의 수 공식을 알고 있었더라면 더 수월했을 것이다.
• combinations 클래스를 사용하려고 했는데, 경우의 수 공식으로 쉽고 빠른 구현이 가능했다.