[백준] 9375 패션왕 신해빈(실버3) - Python

문제

https://www.acmicpc.net/problem/9375
해빈이는 패션에 매우 민감해서 한번 입었던 옷들의 조합을 절대 다시 입지 않는다. 예를 들어 오늘 해빈이가 안경, 코트, 상의, 신발을 입었다면, 다음날은 바지를 추가로 입거나 안경대신 렌즈를 착용하거나 해야한다. 해빈이가 가진 의상들이 주어졌을때 과연 해빈이는 알몸이 아닌 상태로 며칠동안 밖에 돌아다닐 수 있을까?

입력

첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 100이다.

각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.
모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

출력

각 테스트 케이스에 대해 해빈이가 알몸이 아닌 상태로 의상을 입을 수 있는 경우를 출력하시오.

풀이

우선 해빈이는 같은 종류의 의상을 한 번에 2개 이상 입을 수 없다. 그러면 이 문제는 중학교였는지 고등학교 때였는지 수학 시간에 배운 주머니에서 구슬 뽑기와 같은 원리라고 볼 수 있다.

해빈이가 가지고 있는 옷의 종류가 총 3개(상의, 하의, 모자)고, 상의는 2개, 하의는 1개, 모자는 3개라고 가정하자.

그럼 해빈이가 고를 수 있는 상의의 선택지는

1. 안 입는다
2. 상의1
3. 상의2

이렇게 3가지다. 똑같은 방법으로 하의는 2가지, 모자는 4가지의 선택지가 있다.

그럼 해빈이가 입을 수 있는 의상의 조합은 3 * 2 * 4 = 24가지인데, 해빈이가 아무것도 입지 않는 경우의 수는 빼야 하므로 해빈이의 의상 조합은 총 23가지가 된다.

이걸 코드로 작성하면

n = int(input())		# 테스트 케이스 개수

for i in range(n):
    clothes = {}
    m = int(input())	# m = 의상 개수
    
    if m == 0:			# 의상이 없으면 당연히 조합할 수 있는 의상의 가짓수도 0
        print(0)
    else:
        for j in range(m):
            o, c = input().split()
            if c in clothes:
                clothes[c] += 1			# 의상의 종류별로 개수 count -> dict에 저장
            else:
                clothes[c] = 1
        
        total = 1
        for k in clothes:
            total *= (clothes[k] + 1)	# 총 가지수 * (의상 개수 + 안 입는다)

        print(total - 1)				# 총 가지수 - 아무것도 입지 않은 경우의 수