[Python] 백준 9375번 - 패션왕 신해빈

https://www.acmicpc.net/problem/9375

9375번: 패션왕 신해빈

혼자 힘으로 풀었는가? : X (구글에 검색해봄)
알고리즘 유형
- 수학
- 자료 구조
- 조합론
- 해시를 사용한 집합과 맵

문제

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

입력

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

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

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

출력

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

 

힌트

첫 번째 테스트 케이스는 headgear에 해당하는 의상이 hat, turban이며 eyewear에 해당하는 의상이 sunglasses이므로   (hat), (turban), (sunglasses), (hat,sunglasses), (turban,sunglasses)로 총 5가지 이다.

 

---

 

처음에는 dict로 값을 받아서 중복을 허용하지 않는 set에 넣으면서 길이를 구하려 했으나...

 

나의 실력 부족으로 구현해내지 못했고 결국 검색해봤다.

 

검색해보니 이 또한 공식이 있다고 하더이다...

 

(A 옷 개수 + 1) * (B 옷 개수+ 1) * (C 옷 개수+ 1) - 1

 

이게 공식이었던 것이다...

 

공식을 알았으나 어떤 식으로 옷 종류별 개수를 구해야 하나 싶었다.

 

그러고 검색해서 찾은 정답은

 

예를 들어

 

hat headgear
sunglasses eyewear
turban headgear

위의 입력을 받았다면

 

{
	headgear : hat, turban
	eyewear : sunglasses
}

위의 방식이 되도록 set()을 만들어 주면 되는 것이었다.

 

import sys
input = sys.stdin.readline

t = int(input())
for _ in range(t):
    n = int(input())
    data = {}
    for _ in range(n):
        w = list(input().split())
        if w[1] in data:
            data[w[1]].append(w[0])
        else:
            data[w[1]] = [w[0]]
    
    cnt = 1
    for i in data:
        cnt *= len(data[i]) + 1
        
    print(cnt-1)

 

---

위의 코드에서 기본적인 실수를 하여 몇 분간 헤매었다.

 

11번 라인에서

 

data[w[1]].append(w[0])

 

부분에서 계속 오류가 발생했었는데

 

기존에 13번 라인을

 

data[w[1]] = [w[0]] 가 아닌

data[w[1]] = w[0] 으로 선언해서 data[w[1]]의 값이 list 형태가 아닌 str 형태였기 때문에 str 형태에 append가 불가능해 계속 에러가 났었다.

 

찾느라 힘들었다.

 

Ref

https://hongcoding.tistory.com/60

https://pacific-ocean.tistory.com/229

https://velog.io/@ledcost/%EB%B0%B1%EC%A4%80-9375-%ED%8C%8C%EC%9D%B4%EC%8D%AC-%ED%8C%A8%EC%85%98%EC%99%95-%EC%8B%A0%ED%95%B4%EB%B9%88-%EC%8B%A4%EB%B2%843-%EC%A1%B0%ED%95%A9%EB%A1%A0