프로그래머스 (Level 2) - 위장

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. 풀이 접근

1. 의상 종류별로 모아준다. (yellow_hat, green_hat)등 종류가 중요한게 아니라 개수만 필요
   1. face는 몇 개, 모자는 몇 개~
2. 각 의상 종류 + 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와 개수 추출