[프로그래머스] 의상 - ( dict )

 

문제 설명

코니는 매일 다른 옷을 조합하여 입는것을 좋아합니다. 예를 들어 코니가 가진 옷이 아래와 같고, 오늘 코니가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야합니다. 종류 이름 얼굴 동그란 안경, 검정 선글라스 상의 파란색 티셔츠 하의 청바지 겉옷 긴 코트 코니는 각 종류별로 최대 1가지 의상만 착용할 수 있습니다. 예를 들어 위 예시의 경우 동그란 안경과 검정 선글라스를 동시에 착용할 수는 없습니다. 착용한 의상의 일부가 겹치더라도, 다른 의상이 겹치지 않거나, 혹은 의상을 추가로 더 착용한 경우에는 서로 다른 방법으로 옷을 착용한 것으로 계산합니다. 코니는 하루에 최소 한 개의 의상은 입습니다. 코니가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

 

 

풀이

def solution(clothes):
    dict1 = {}
    num = []
    an=1
    print(type(clothes))
    for i in clothes:
        if i[1] in dict1:
            dict1[i[1]] += 1
        else:
            dict1[i[1]] =1
    
    num = list(dict1.values())
    
    for i in range(len(num)):
        an *= num[i]+1
        
    return an-1

 

종류가 5개 있는 것 같지만 예제에서 headgear, eyewear, face만 줘서 정확하게 파악하긴 힘들었다.

그래서 dict 배열을 만들어 새로운 종류가 나올 때마다 추가할 수 있게 하였다. 

 

종류가 3개인 [ 얼굴 - 2개, 상의 - 1개, 하의 - 1 ] 개가 있다면

(2+1)(1+1)(1+1)-1 = 11 개가 된다.

1 - 얼굴(1)

2 - 얼굴(2)

3 - 상의(1)

4 - 하의(1)

5 - 상의(1) 하의(1)

6 - 얼굴(1) 상의(1)

7 - 얼굴(1) 하의(1)

8 - 얼굴(1) 상의(1) 하의(1)

9 - 얼굴(2) 상의(1)

10 - 얼굴(2) 하의(1)

11 - 얼굴(2) 상의(1) 하의(1)

여기서의 규칙은 약수의 개수를 찾는 것과 마찬가지로 원소의 포함여부를 곱하여 연산한다.

얼굴(1), 얼굴(2) :  x포함, 1개만 포함 -> 3종류

상의(1)      : x포함, 1개 포함 -> 2종류

하의(1)      : x포함, 1개 포함  -> 2종류

모두 안 입는 경우 제외

 

 

출처

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

프로그래머스