[Beakjoon] 9375: 패션왕 신해빈 - python

문제

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

 

▶️ 입력 조건 

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

▶️ 출력 조건

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

---

풀이

경우의 수 공식을 이용해서 문제를 해결할 수 있다.

n개의 의상이 있을 경우 모든 경우의 수는 아래와 같은 공식으로 구할 수 있다.

 

(1 + 1) * (2 + 1) *....(n + 1) - 1

 

n에 +1을 해주는 이유는 해당 종류 의류를 안입는 경우를 추가해주기 위함이고

마지막 최종합에서 -1을 해주는 이유는 아무것도 입지 않은 경우는 경우의 수에서 제외하기 위함이다.

알몸으로는 돌아다닐 수 없다고 문제에 명시 되어있음

 

이를 기반으로 dict에 각 의상 종류의 count 값을 저장 한 후 계산하는 것으로 문제를 해결했다.

참고로 collections의 Counter 클래스를 사용하면 더 쉽게 해결할 수 있다.

그러나 문명의 이기에 편해지지 않기 위해 직접 구현함...

import sys

input = sys.stdin.readline
t = int(input().rstrip())

for _ in range(t):
    n = int(input().rstrip())
    costumes_count = {}
    
    for _ in range(n):
        _, category = input().rstrip().split()
        
        if category in costumes_count.keys():
            costumes_count[category] += 1
        else:
            costumes_count[category] = 1
    
    count = 1
    
    for key in costumes_count.keys():
        count *= costumes_count[key] + 1
        
    print(count-1)