[백준] #9375 패션왕 신혜빈(python)

📌문제

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

입력

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

각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.
다음 n개에는 해빈이가 가진 의상의 이름과 의상의 종류가 공백으로 구분되어 주어진다. 같은 종류의 의상은 하나만 입을 수 있다.
모든 문자열은 1이상 20이하의 알파벳 소문자로 이루어져있으며 같은 이름을 가진 의상은 존재하지 않는다.

출력

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

예제 입력

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

예제 출력

5
3

힌트

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

백준 9375번 문제

💡Idea

어딘가에서 비슷한 문제를 풀었던 기억이 나는 문제입니다🤔
처음 풀 때는 꽤 오랜 시간이 걸렸던 것 같은데, 지금은 몇 분 걸리지 않은 것을 보아 실력이 조금씩 성장하고 있는 것을 느낄 수 있는 듯 합니다😋

어쨋뜬,

이 문제는 종류별 옷👚을 조합하는 문제입니다.
의상을 입을 수 있는 경우는 아래와 같이 생각할 수 있습니다.

(A 종류의 옷 가짓수 + 1) $\times$ (B 종류의 옷 가짓수 + 1) $\times$ $...$ $-1$

같은 종류의 의상은 하나만 입을 수 있지만, 대신 아예 입지 않을 수도 있습니다.

따라서, 해당 종류를 선택하지 않은 경우를 고려하기 위하여 각 종류의 옷 가짓수에 각각 1을 더해줍니다.

그렇게 1을 더한 값들을 모두 곱해주면 해당 종류의 옷들로 입을 수 있는 의상의 조합 수가 나옵니다.

대신, 모든 종류의 옷을 선택하지 않은 경우도 존재할 수 있으니 1을 빼주어야 합니다.

💻코드

• ⏰ 시간 : 96 ms / 메모리 : 32424 KB

import sys
from functools import reduce

input = sys.stdin.readline
t = int(input())

for _ in range(t):
    n = int(input())
    clothes = {} 
    for __ in range(n):
        cloth, kind = input().split()
        if kind in clothes: clothes[kind] += 1 
        else: clothes[kind] = 1
    print(reduce(lambda ans, x: ans * (x + 1),clothes.values(), 1) - 1)

📝코드 설명

변수

• clothes : 각 옷 종류의 가짓수를 담는 딕셔너리로, {옷의 종류 : 해당 종류의 옷의 가짓수} 형태로 되어 있습니다.

저는 딕셔너리를 사용하여 구현하였습니다.

입력을 통하여 옷의 종류와 각 종류의 옷의 가짓수를 저장하고, 그 개수들에 1씩 더한 뒤 모든 값을 곱해주었습니다.

마지막으로, 1을 빼주어 조합의 개수를 계산할 수 있도록 하였습니다.

또한, 각 리스트에 담긴 값에 1을 더하고 그 값들을 곱해준 최종 누적값을 한 줄로 얻기 위하여 reduce를 사용해주었습니다.