1. 문제 설명
스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장한다.
- 스파이가 가진 의상들이 담긴 2차원 배열 clothes
- 서로 다른 옷의 조합의 수를 return
2. 제한사항
- clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있다.
- 스파이가 가진 의상의 수는 1개 이상 30개 이하이다.
- 같은 이름을 가진 의상은 존재하지 않는다.
- clothes의 모든 원소는 문자열로 이루어져 있다.
- 모든 문자열의 길이는 1 이상 20이하인 자연수이고 알파벳 소문자 또는 '_'로만 이루어져 있다.
- 스파이는 하루에 최소 한 개의 의상은 입는다.
3. 입출력 예
clothes | return |
[["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]] | 5 |
[["crow_mask", "face"], ["blue_sunglasses", "face"], ["smoky_makeup", "face"]] | 3 |
- 첫 번째 경우 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
4. 풀이 접근
- 의상 종류별로 모아준다. (yellow_hat, green_hat)등 종류가 중요한게 아니라 개수만 필요
- face는 몇 개, 모자는 몇 개~
- 각 의상 종류 + 1씩 더해서 곱셈해서 마지막에 1만 빼준다. (스스로 입은거 까지 포함)
5. 코드
import collections
def solution(clothes):
answer = 1 #곱셈을 위한
result = []
for i in range(len(clothes)):
result.append(clothes[i][1]) #1 개수만 필요하므로 종류별로 모아준다.
a = collections.Counter(result) #1 종류별 개수를 구해준다.
for key, value in a.items():
answer *= (value + 1)
return answer - 1 #2
- [5~7] 과정을 하나의 코드로 간결하게 정리할 수 있어서 바꿔봤다.
- value값 기준으로 Counter을 세주는 ( cat for value, cat in clothes)를 사용한다.
import collections
def solution(clothes):
answer = 1
cloth = collections.Counter(cat for value, cat in clothes)
for key, value in cloth.items():
answer *= (value + 1)
return answer - 1
6. 알아둘만한 문법
import collections
clothes = ["a","b","c","a"]
cloth = collections.Counter(clothes)
print(cloth)
#출력 결과
Counter({'a': 2, 'b': 1, 'c': 1})
- 1차원 리스트에서 key와 value 개수 추출
import collections
clothes = [["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]
cloth = collections.Counter(cat for value, cat in clothes)
print(cloth)
#출력 결과
Counter({'headgear': 2, 'eyewear': 1})
- 2차원 리스트에서 value값 기준으로 value와 개수 추출
'CodingTest > 해시 알고리즘' 카테고리의 다른 글
프로그래머스 (Level 2) - 전화번호 목록 (0) | 2022.08.06 |
---|---|
프로그래머스 (Level 1) - 완주하지 못한 선수 (0) | 2022.08.06 |
프로그래머스 (Level 1) - 폰켓몬 (0) | 2022.07.29 |