[알고리즘] 프로그래머스 알고리즘 고득점 kit - [Python] 의상

문제 설명

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

- 코니는 각 종류별로 최대 1가지 의상만 착용할 수 있습니다. 예를 들어 위 예시의 경우 동그란 안경과 검정 선글라스를 동시에 착용할 수는 없습니다.
- 착용한 의상의 일부가 겹치더라도, 다른 의상이 겹치지 않거나, 혹은 의상을 추가로 더 착용한 경우에는 서로 다른 방법으로 옷을 착용한 것으로 계산합니다.
- 코니는 하루에 최소 한 개의 의상은 입습니다.코니가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

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

 

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/42578

프로그래머스

풀이 방법

이 문제의 키 포인트는 코니의 의상을 통하여 만들 수 있는 조합의 수를 어떻게 구하는지 알아야한다는 것입니다.

얼굴	안경, 선글라스
상의	티셔츠

 

만일, 코니가 가지고 있는 의상이 위와 같다고 하면 얼굴에서는 안경을 고르거나, 선글라스를 고르거나, 고르지 않을 수 있습니다.또한, 상의에서도 티셔츠를 고르거나 고르지 않을 수 있습니다.

 

얼굴 = [안경, 선글라스] / 상의 = [티셔츠] 와 같은 배열로 본다면 얼굴에서 고를 수 있는 수는 '얼굴 배열의 길이 + 1'입니다.

 

물론, 얼굴에서 고르지 않고, 상의에서도 고르지 않는 경우(코니가 아무것도 입지 않는 경우)는 없다고 하였으니 최종 결과에서 1을 빼주면 됩니다.

 

코드 설명

1. 의상의 종류를 key로 하고 의상의 이름을 value로 갖는 딕셔너리를 하나 생성합니다. clothes를 순회하며 이를 초기화합니다.

2. values() 를 통하여 딕셔너리의 value 배열을 얻고, 이를 리스트로 변환합니다.

3. 배열 values를 순회하며 answer에  (value 배열의 길이 + 1)을 곱해준 뒤, 1을 제거하여 정답으로 반환합니다.

 

코드 전문

import itertools

def solution(clothes):
    dic = {}
    answer = 1
    
    for cloth in clothes:
        key = cloth[1]
        value = cloth[0]
        if not key in dic:
            dic[key] = []
        
        dic[key].append(value)
        
    values = list(dic.values())
    
    for value in values:
        answer *= (len(value) + 1)
    
    # 아무것도 선택하지 않는 경우
    answer -= 1

    return answer