[백준 알고리즘] 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

---

문제 접근

한 종류의 옷을 두 개 이상 입을 수 없다는 조건이 중요하다. (예를 들면 모자 두개를 겹쳐 쓸 수 없는 것)

얼핏 보면 조합의 개수를 구하는 문제 같아 복잡해 보이지만,

eyewear의 개수 4개, 모자 2개, 상의 3개를 보유하고 있다고 가정했을 때, 4 * 2 * 3 = 각 카테고리별로 1개씩 고르는 경우의 수이다.

이를 이용하여 문제를 풀면 될 것 같다.

 

정답 코드

import sys

def test(n):
    if n == 0:                   #옷 개수가 0인 경우엔 0을 리턴하고 종료
        return 0
    dict = {}
    count = 1                    #곱연산을 해야하므로 0이 아닌 1로 설정
    for i in range(n):
        item, category = sys.stdin.readline().rstrip().split()
        if category not in dict: #dictionary에 카테고리와 그 개수를 추가하는 부분
            dict[category] = 1
        else: dict[category] += 1

    if len(dict) > 1:            #카테고리가 2개 이상일 때(여러 종류의 옷)
        for i in list(dict.values()):
            count *= (i + 1)     #i개 중의 하나를 고르는 경우의 수(iC1) = i ... + 하나도 고르지 않는 경우의 수 = 1
        return count - 1         #모든 옷들을 하나도 고르지 않는 경우의 수는 문제 조건 위배이므로 뺀다.
    else:                        #카테고리가 1개일 때(옷이 한종류)
        return list(dict.values())[0]

t = int(sys.stdin.readline())
for i in range(t):
    print(test(int(sys.stdin.readline().rstrip())))

 

옷의 종류별로 보유한 옷들의 수를 딕셔너리에 저장한다. ex: {eyewear: 3, headgear: 2}

딕셔너리의 values()를 list로 만들어 for문으로 돌면서 카운트 해준다.

옷 종류가 2개 이상일 경우에는,

i개 중 하나를 고르는 경우의 수인 i와 하나도 고르지 않는 경우의 수 1을 더하고 count 변수에 곱연산 해준다.

리턴할 때는 -1을 해야 한다. (모든 옷 종류를 하나도 고르지 않은 경우를 제외)

 

주의

if n == 0:                   #옷 개수가 0인 경우엔 0을 리턴하고 종료
    return 0

테스트 케이스에는 없지만,

• 각 테스트 케이스의 첫째 줄에는 해빈이가 가진 의상의 수 n(0 ≤ n ≤ 30)이 주어진다.

라는 조건에 의해 아예 의상 수가 0인 경우를 처리해주지 않는다면 런타임 에러가 발생하니 주의해야 한다.

---

9375번: 패션왕 신해빈