[알고리즘] 문제 풀이

[해시] 프로그래머스 - 위장

yeon_zoo 2021. 10. 29. 16:18

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42578

 

문제 설명:

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

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

종류 이름
얼굴 동그란 안경, 검정 선글라스
상의 파란색 티셔츠
하의 청바지
겉옷 긴 코트

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

 

제한사항

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

입출력 예

clothes return
[["yellowhat", "headgear"], ["bluesunglasses", "eyewear"], ["green_turban", "headgear"]] 5
[["crowmask", "face"], ["bluesunglasses", "face"], ["smoky_makeup", "face"]] 3

 

문제 풀이: 

먼저 해시 자료구조로 입력을 다음과 같이 바꿔야 한다고 생각했다.

headgear : yellowhat, green_turban

eyewear : bluesunglasses

이런 구조이다. 그 다음엔 가능한 조합 수를 구하는 과정이었는데, 각 옷 종류에 원소 개수 n에서 아무것도 선택하지 않는 선택지를 추가한 n+1 에서 한 가지를 선택하는 경우를 생각하면 되었다. 그리고 그 값을 다 곱해주면 된다. 주의해야 할 점은 모든 옷 종류에서 아무것도 선택하지 않는 선택을 한 경우의 한 가지를 빼주어야 했다. 그래서 다음의 경우에서는 headgear 선택하는 경우의 수 3 * eyewear에서 선택하는 경우의 수 2 - 1 (모든 종류에서 아무것도 선택하지 않았을 때) = 5 가지임을 확인할 수 있었다.

def solution(clothes):
    categories = {}	# 종류별 옷을 담을 해시 구조
    answer = 1
    for cloth in clothes:
        if cloth[1] in categories.keys():
            categories[cloth[1]].append(cloth[0])
        else:
            categories[cloth[1]] = [cloth[0]]
    for category in categories.values():
        answer *= (len(category) +1)
    return answer-1