문제 설명
스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.
예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.
종류 이름
얼굴 동그란 안경, 검정 선글라스
상의 파란색 티셔츠
하의 청바지
겉옷 긴 코트
스파이가 가진 의상들이 담긴 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
입출력 예 설명
예제 #1
headgear에 해당하는 의상이 yellow_hat, green_turban이고 eyewear에 해당하는 의상이 blue_sunglasses이므로 아래와 같이 5개의 조합이 가능합니다.
-
yellow_hat
-
blue_sunglasses
-
green_turban
-
yellow_hat + blue_sunglasses
-
green_turban + blue_sunglasses
예제 #2
face에 해당하는 의상이 crow_mask, blue_sunglasses, smoky_makeup이므로 아래와 같이 3개의 조합이 가능합니다. -
crow_mask
-
blue_sunglasses
-
smoky_makeup
먼저 각 아이템별로 category를 만든다.-> 각 카테고리(모자, 안경, 상의 등)별로 아이템의 갯수를 구한다.
[c for _ , c in clothes]
-> 0 번째 인덱스인 의상 이름을 전부 언더바 처리해서 버렸으므로,
다음과 같이 '의상의 종류'만 모여있는 리스트가 생성된다.
[headgear, eyewear, headgear]
그렇다면 ,
Counter([c for _, c in clothes])
위 코드는 다음과 같이 표현될 수 있다.
Counter(['headgear', 'eyewear', 'headgear'])
이 코드가 실행되면 다음과 같은 Counter 객체를 반환한다.
Counter({'headgear': 2, 'eyewear': 1})
그리고 각 아이템별로 경우의 수를 구하면 되겠지만
예를들어 선글라스만 끼는 경우라던지, 코트만 거치는 경우라던지 각 아이템을 하나씩만 걸치는 경우의 수(제한 사항: 스파이는 하루에 최소 한 개의 의상은 입습니다.)를 위해(근데 상식적으로 이런 경우가 말이 된다고 생각하는지..?? ..... 에휴) +1(ex. headgear : 2개 + headgear를 입지 않는 경우의 수 한개..)을 더해주어야 한다.
그리고 마지막으로 아무것도 입지 않는 경우를 빼주면 된다.
from collections import Counter
def solution(clothes):
answer = 1
category = Counter([c for _, c in clothes])
for i in category:
answer *= (category[i]+1)
return answer-1
22.01.29 복습
