[프로그래머스/python] 위장

위장

https://programmers.co.kr/learn/courses/30/lessons/42578

코딩테스트 연습 - 위장

 

---

문제 설명

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

 

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

 

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

---

제한사항

• clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
• 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
• 같은 이름을 가진 의상은 존재하지 않습니다.
• clothes의 모든 원소는 문자열로 이루어져 있습니다.
• 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
• 스파이는 하루에 최소 한 개의 의상은 입습니다.

---

입출력 예

clothes	return
[["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]	5
[["crow_mask", "face"], ["blue_sunglasses", "face"], ["smoky_makeup", "face"]]	3

 

---

풀이 과정

 

분류가 a, b, c 가 있을 때 풀이 과정은 다음 공식과 같습니다.

 

( a분류인 원소 개수 + 1 )*( b분류인 원소 개수 + 1 )*( c분류인 원소 개수 + 1 )

 

예를 들어

 

분류의 종류가 'fruit' 와 'animal' 이 있다고 가정합니다.

 

'fruit' 의 원소는 'apple', 'banana', 'melon'이 있고

 

'animal' 의 원소는 'rabbit', 'tiger'가 있습니다.

 

그러면 존재하는 경우의 수는 

 

apple

banana

melon

rabbit

tiger

apple, rabbit

apple, tiger

banana, rabbit

banana, tiger

melon, rabbit

melon, tiger

 

총 11가지입니다.

 

이번에는 fruit의 원소가 3개이고 animal의 원소가 2개일때 모든 분류를 1개씩만 포함되도록 하는 경우의 수는 3 * 2 입니다.

 

apple, rabbit

apple, tiger

banana, rabbit

banana, tiger

melon, rabbit

melon, tiger

 

이번에는 모든 분류에 x 원소 를 추가하여 경우의 수를 구해봅시다.

 

그러면 fruit의 원소는 ['apple', 'banana', 'melon', 'X'] 이고,

 

animal의 원소는 ['rabbit', 'tiger', 'X'] 입니다.

 

같은 원리로 모든 분류를 1개씩만 포함되도록 하는 경우의 수는 4 * 3 이 될것입니다.

 

 

apple, rabbit

apple, tiger

apple, X

banana, rabbit

banana, tiger

banana, X

melon, rabbit

melon, tiger

melon, X

X, rabbit

X, tiger

X, X

 

여기서 X를 제외하면 우리가 구하는 답이 됩니다.

 

문제에서는 아무것도 입지 않은 경우는 제외하라고 명시되어 있기 때문에

 

공식은 4 * 3 - 1 이 됩니다.

 

풀이

 

def solution(clothes):
    dic = {}
    answer = 1
    
    for co in clothes:                      # 분류를 하여 딕셔너리에 저장합니다.
        if co[1] in dic.keys():
            dic[co[1]].append(co[0])

        else :
            dic[co[1]] = [co[0]]

    for d in dic.values():                  
        answer *= len(d) + 1

    return answer-1

 

 

---

 사용된 개념

 

* in 연산자

 

* 딕셔너리에 리스트 추가하기