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

문제 정보

제목 : 패션왕 신해빈

번호 : 9375번

사용 언어 : Python

문제 링크

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

 

시간 제한	메모리 제한
1 초	128 MB

 

문제

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

 

입력

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

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

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

 

출력

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

 

입출력 예제

---

나의 풀이

옛날에 프로그래머스에서 풀었던 문제와 유사한 문제였습니다. 테스트 케이스의 수를 입력받아 그만큼 반복하는 for루프를 생성합니다. answer에 1을 저장하고 의상의 수를 입력받아 반복합니다. 반복문의 내부에서 input()함수를 통해 입력받고 문자열을 분리하여 의상 이름과 종류로 나누어서 저장합니다. 그다음 딕셔너리에 의상의 종류를 키값으로 하여 이름을 추가합니다. 딕셔너리의 값에 접근하는 for루프를 생성하여 answer에 len(v) + 1을 합니다. 이때 +1을 하는 이유는 해당 종류의 의상을 안입는 경우를 생각해줘야하기 때문입니다. 그리고 answer을 출력하는데 옷을 아무것도 안입는 경우는 없으므로 -1을 출력합니다. 

 

코드

from collections import defaultdict

wardrobe = defaultdict(list)
test_case = int(input())

for i in range( test_case ) :
    answer = 1
    clothes_cnt = int(input())
    for j in range(clothes_cnt) :
        clothes,cType = input().split()
        wardrobe[cType].append(clothes)
    
    for v in wardrobe.values() :
        answer *= len(v) + 1

    print(answer - 1)
    wardrobe.clear()