[Python] 백준 9375 패션왕 신혜빈

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

9375번: 패션왕 신해빈

 

문제

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

 

입력

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

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

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

 

출력

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

 

예제 입력 1 

2
3
hat headgear
sunglasses eyewear
turban headgear
3
mask face
sunglasses face
makeup face

 

예제 출력 1

5
3

 

코드 및 설명

n=int(input())
for i in range(n):
    m=int(input())
    ctype={}#각 타입 수 저장 딕셔너리
    for j in range(m):
        name,type=input().split()
        if type in ctype:
            ctype[type].append(name)
        else:
            ctype[type]=[name]
    cnt=1

    for k in ctype:
        cnt*=(len(ctype[k])+1)
    print(cnt-1)

오랜만에 딕셔너리를 이용한 문제.

같은 거를 처리해주어야 하기 때문에 리스트보다는 딕셔너리가 낫다고 판단했는데 그 방식이 헷갈렸던 문제

처음에는 []로 해서 추가해주고 안에 키가 존재한다면 append를 통해 ()에 변수를 넣어서 처리를 해 주어야 한다.

그리고 경우의 수를 구하는 식이 중요한데

 

식은

각 종류 키의 길이+1을 해서 다 곱해주어야 한다.

이 이유는 하나만 입어도 된다고 했기 때문.

그리고 알몸인 상태는 빼주어야 하므로 전체에서 -1을 해주기

 

마지막으로 그 전체 횟수를 출력해주면 끝!

 

이 문제 역시 각 시도마다 출력을 해주어야 해서 for문 안에서 처리를 해주었다.