[프로그래머스-Lv2] 위장 / Python

코딩테스트 연습 - 위장

import sys

input = sys.stdin.readline

clothes = [list((map(str, input().split()))) for _ in range(int(input()))]


def solution(clothes):
    temp = {}
    for i in clothes:
        if i[1] in temp:
            temp[i[1]] += 1
        else:
            temp[i[1]] = 1
    answer = 1
    for i in temp.values():
        answer *= i + 1
    return answer - 1


print(solution(clothes))

 

해설

딕셔너리를 이용해 옷 종류의 개수를 저장했다. 구한 값으로 입을 수 있는 옷의 조합 수를 구하는 공식은 (각 종류 +1)을 서로 곱해주고 결과에 하나를 빼준 값이다.