[baekjoon] 백준 9375번 : 패션왕 신혜빈 (by python) 딕셔너리

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

9375번: 패션왕 신해빈

 

정답 code

#패션왕 신해빈
import sys
input = sys.stdin.readline

t = int(input())


for _ in range(t):
    n = int(input())
    clo = {}
    for _ in range(n):
        wear = list(input().split())
        if wear[1] in clo:
            clo[wear[1]].append(wear[0])
        else:
            clo[wear[1]] = [wear[0]]

    cnt = 1
    for k in clo:
        cnt *=(len(clo[k])+1)

    print(cnt-1)

 

solution

딕셔너리를 사용하여  같은 종류의 의상끼리 정리했다.

이미 있는 종류라면 append를 해주고 없다면 새로 추가해준다.

 

알몸이 아닌 한가지 이상의 의상을 입는 경우의 수는

(같은종류 의상수+1)*(같은종류 의상수+1).... -1이 된다.