[🥈3 / 백준 9375 / 파이썬] 패션왕 신해빈

 

9375번: 패션왕 신해빈

 

문제

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

입력

첫째 줄에 테스트 케이스가 주어진다. 테스트 케이스는 최대 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

 

 

풀이

 

일반 경우의 수 문제이다.

 

옷의 종류마다 하나씩 선택할 수 있는데, 선택을 하지 않을 수도 있다.

 

따라서 옷이 상의 3벌, 하의 3벌이 있다면 선택을 하지 않는 경우까지 포함하여 각각 4가지씩의 경우가 생긴다.

 

4 * 4 = 16

 

단, 알몸인 경우는 제외하므로 -1을 해주어 조건에 맞추자.

 

 

코드 작성 요령

 

처음엔 옷의 이름과 종류가 주어지므로, dict 를 활용할까 고민했었다.

 

하지만 어차피 문제에서 같은 이름을 가진 의상은 주어지지 않는다고 했으므로

 

이름은 제외하고 같은 종류의 옷이 몇 번 들어왔는지만 체크해주면 된다.

 

import sys
input = sys.stdin.readline

for i in range(int(input())):
    wlist = []   # 옷의 종류 
    wcount = []  # 종류별 옷의 개수

    for i in range(int(input())):
        a,w = input().split()

        if w in wlist:
            wcount[wlist.index(w)] += 1
        else:
            wlist.append(w)
            wcount.append(2)
            # 입지 않는 경우도 있으니 
            # 1이 아닌 2를 추가해준다.

    result = 1
    for i in wcount:
        result *= i
    print(result - 1)
    # 알몸인 경우 제외