[알고리즘] 문제 풀이
[해시] 프로그래머스 - 위장
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