백준 9375번. 패션왕 신해빈 (Python / 파이썬)

문제

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

입력

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

출력

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

---

문제 풀이

해빈이가 가진 의상들이 주어졌을 때 

해빈이가 입을 수 있는 의상 조합의 경우의 수를 구하는 문제이다. 

이때, 같은 종류의 의상은 하나만 입을 수 있다. 

 

만약, 아래 표와 같이 의상들이 주어졌을 때

이름	hat	sunglasses	turban
종류	headgear	eyewear	headgear

종류가 headgear인 의상은 hat, turban이고 eyewear인 의상은 sunglasses이다. 

따라서 경우의 수는 (hat), (turban), (sunglasses), (hat,sunglasses), (turban,sunglasses)로 총 5가지이다. 

 

즉, 각 종류에 해당하는 의상의 수를 세고

headgear = 2개 eyewear = 1개 경우의 수를 구해보면 종류가 headgear인 의상을 입는 경우의 수 * 종류가 eyewear인 의상을 입는 경우의 수 - 1 이다. 

(2 + 1) * (1 + 1) - 1 = 5가지 

이때 +1을 해주는 이유는 해당 종류의 의상을 입지 않는 경우를 고려해준 것이다. 

또한 마지막에 -1을 해주는 이유는 아무것도 입지 않는 경우를 빼준 것이다. 

 

따라서 각 종류에 해당하는 의상의 수를 세어 경우의 수를 구하는 식을 세워주면 답을 출력할 수 있다. 

각 종류에 해당하는 의상의 수를 세는 방법을 

dictionary와 Counter 라이브러리를 이용해 각각 구현해보았다. 

---

My Code 1 - dictionary를 이용한 방법 

for tc in range(int(input())):
    n = int(input()) # 의상의 수 

    wear = {} # 각 종류의 의상 수를 저장할 dict
    for _ in range(n):
        x = input().split()
        if x[1] not in wear: # 저장되어 있지 않다면 1 저장
            wear[x[1]] = 1 
        else:                # 이미 저장되어 있다면 개수 늘려줌 
            wear[x[1]] += 1 

    res = 1
    for i in wear:
        res *= wear[i] + 1  # 각 의상을 입는 경우의 수를 곱해줌 
    
    print(res - 1) # 아무것도 입지 않는 경우 빼서 출력

My Code 2 - Counter를 이용한 방법 

from collections import Counter

for tc in range(int(input())):
    n = int(input()) # 의상의 수 

    wear = []
    for _ in range(n):
        x = input().split()
        wear.append(x[1]) # 의상의 종류를 배열에 저장 
    
    new_wear = Counter(wear) # 각 종류에 해당하는 의상의 수를 세어줌 

    res = 1
    for i in new_wear:
        res *= new_wear[i] + 1 # 각 의상을 입는 경우의 수를 곱해줌
    
    print(res - 1) # 아무것도 입지 않는 경우 빼서 출력