[python/파이썬] 백준 9375 패션왕 신해빈

문제

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

입력

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

• 각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
• 다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.

모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

출력

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

예제 입력 1 복사

2
3
hat headgear
sunglasses eyewear
turban headgear
3
mask face
sunglasses face
makeup face

예제 출력 1 복사

5
3

 

 

[소스 코드]

 

수학적으로 접근해야하는 문제이다.

만일 바지 3종류, 상의 2종류, 모자 2종류가 있다고 한다면

 

가능한 경우의 수는

1. 옷을 1가지만 입는 경우

    3 + 2 + 2 = 7  #바지 3 + 상의 2 + 모자 2

 

2. 옷을 2가지 조합해 입는 경우

    (3*2) + (3*2) + (2*2)  = 6 + 6 + 4 = 16   #바지, 상의 / 바지, 모자 / 상의, 모자 

 

3. 옷을 3가지 모두 조합해 입는 경우

    3*2*2 = 12  #바지, 상의 모자

 

이 3가지 경우를 모두 더해야 한다.

35가지가 나온다.

---

항상 모든 옷을 입어야 하는 것은 아니기에 입는 경우와 입지 않은 경우로 나눠서 계산하면

(3+1) * (2+1) * (2+1) = 36

#바지 3종류 + 바지를 입지 않은 경우 / 상의 2종류 + 상의를 입지 않은 경우 / 모자 2종류 + 모자를 쓰지 않은 경우

여기에서 모두 입지 않은 경우, 즉 나체일 때를 빼주면 위에서 구한 값과 동일하게 35가 나온다.

 

---

옷 종류에 따른 가짓수를 A, B, C, D,...라고 할 때

가능한 경우의 수를 식으로 나타내면 아래와 같다.

 

(A+1) * (B+1) * (C+1) * (D+1) +......+......-1

 

이 식을 이용해서 소스 코드를 작성하였다.

 

import sys
input = sys.stdin.readline

T = int(input())

for _ in range(T):
    n = int(input())
    
    clothes = {}
    for i in range(n):
        a,b = map(str, input().split())
        
        if b not in clothes.keys():
            clothes[b] = 1
        else:
            clothes[b] += 1

    sum = 1
    for key in clothes.keys():
        sum *= clothes[key] + 1

    print(sum - 1)