본문 바로가기
백준

[백준] 9375번 파이썬 (패션왕 신해빈)

by 헤이즐넛 좋아하는 개발자 2024. 3. 29.

1. 문제와 예제

(문제링크: https://www.acmicpc.net/problem/9375)

 

9375번: 패션왕 신해빈

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

www.acmicpc.net

문제
예제


2. 전체 코드

import sys
input = sys.stdin.readline

# 입력 받기
t = int(input().rstrip()) # 테스트 케이스 수

for _ in range(t):
    n = int(input().rstrip()) # 의상 수
    dct = {}
    for _ in range(n):
        # ex. dct = {'headgear': 2, 'eyewear': 1}
        name, type = input().split() # 의상 이름, 종류
        if type in dct:
            dct[type] += 1
        else:
            dct[type] = 1
        
    res = 1
    for k, v in dct.items():
        res *= (1+v)
    res -= 1
    
    # 결과 출력하기
    print(res)

3. 코드 해설

우리는 옷의 종류 각각에 대한 개수만 필요하므로 dct 구조로 바꿨고 value에 개수가 들어가도록 했다. 그 후 정답이 나오는 규칙을 찾아 코드로 작성했다.