[백준/Python] 9375.패션왕 신해빈

문제

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

입력

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

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

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

출력

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

 

 

 

 

해설:

 

되게 간단한데 생각 없이 풀어서 메모리 초과가 난 문제..

 

예를 들어

 

해빈이가 갖고 있는 옷이 종류별로 a, b, c라 하자.

 

해빈이가 한 종류의 옷만 입는 경우

 

a + b + c

 

두 종류의 옷을 입는 경우

 

ab + bc + ac

 

세 종류의 옷을 입는 경우

 

abc

 

인데 이걸 합쳐보면

 

a + b + c + ab + bc + ac + abc

 

이걸 묶어버리면...

 

(a+1)(b+1)(c+1) - 1

 

홀리 쓋...

 

이걸 생각 못했네 ㅋ....

 

import sys
import collections
 
T = int(sys.stdin.readline())
for _ in range(T):
    total = 1
    cases = collections.defaultdict(int)
    n = int(sys.stdin.readline())
    for _ in range(n):
        a, b = sys.stdin.readline().split()
        cases[b] += 1
 
    cases = list(cases.values())
    for i in range(len(cases)):
        total *= (cases[i]+1)
    print(total-1)

 

알아서 코드 이해하세요 ㅎ