[파이썬] 프로그래머스 Level.2 | 해시 | 위장
문제 설명
스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.
예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.
종류이름
| 종류 | 이름 |
|
얼굴 |
동그란 안경, 검정 선글라스 |
|
상의 |
파란색 티셔츠 |
|
하의 |
청바지 |
|
겉옷 |
긴 코트 |
스파이가 가진 의상들이 담긴 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 |
입출력 예 설명
예제 #1
headgear에 해당하는 의상이 yellow_hat, green_turban이고 eyewear에 해당하는 의상이 blue_sunglasses이므로 아래와 같이 5개의 조합이 가능합니다.
1. yellow_hat 2. blue_sunglasses 3. green_turban 4. yellow_hat + blue_sunglasses 5. green_turban + blue_sunglasses
예제 #2
face에 해당하는 의상이 crow_mask, blue_sunglasses, smoky_makeup이므로 아래와 같이 3개의 조합이 가능합니다.
1. crow_mask 2. blue_sunglasses 3. smoky_makeup
풀이과정
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
def solution(clothes):
answer = 1
hashclothes={}
for one in clothes:
if one[1] not in hashclothes:
hashclothes[one[1]]=1
elif one[1] in hashclothes:
hashclothes[one[1]]+=1
clothes_num=list(hashclothes.values())
for c in clothes_num: #조합수 구하는 식
answer*=(c+1)
answer-=1
return answer
|
이 문제는 수학지식이 필요했던 문제였다.
총 조합의 수를 구할 때, 만약 a,b,c 3개의 조합을 구한다면, (a+1)(b+1)(c+1)-1 이 조합의 수를 구하는 식이다!
이 지식은 '질문하기' 게시판에서 나와 같은 고민을 하는 사람들에게 댓글로 정보가 주어져 있어서 알게 되었다.
이 조합 구하는 식을 몰랐다면 상당히 헤맬 것이다. 초반에 나도 그랬었고..
문제 푼 방법을 쓰기 전에, 일단 해시 문제에서는 어느 변수를 Key로 하고, 어느 변수를 value로 할지를 먼저 생각해야 한다.
처음에 이 문제를 접했을 땐, 해시에 대해 잘 몰랐다. 그래서 막연하게 key->value라는 직관적인 생각으로, key에 옷 이름, value에 옷 종류를 넣을까 했지만, key는 고유한 값이라는 것을 구글링을 통해 배웠다.
그리고 이 문제에서는 옷 이름들은 중요하지 않고, 옷 종류별 '수'가 중요하다는 걸 깨달았다.
따라서 key: 옷 종류, value: 그 종류의 옷 수 로 결정을 짓고, 코드를 짰다.
이 문제는 내가 해시를 이해하는 데에 상당히 도움이 되었던 문제였다.
처음엔 해시의 개념도 잘 몰랐는데, 문제를 풀기 위해 구글링을 하면서 해시와 관련된 함수들, 해시의 특징들을 많이 배웠다.
엄청나게 어려운 문제들보다 초보자에겐 이런 적당한 난이도의 문제가 실력 향상에 도움이 되는 것 같다!