[프로그래머스] 위장

문제 링크 >> https://programmers.co.kr/learn/courses/30/lessons/42578

 

 

📋 문제

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

 

종류	이름
얼굴	동그란 안경, 검정 선글라스
상의	파란색 티셔츠
하의	청바지
겉옷	긴 코트

 

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

 

 

👉 입력

• clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
• 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
• 같은 이름을 가진 의상은 존재하지 않습니다.
• clothes의 모든 원소는 문자열로 이루어져 있습니다.
• 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
• 스파이는 하루에 최소 한 개의 의상은 입습니다.

 

 

👈 출력

 

 

📝 풀이

 

def solution(clothes):
    answer = 1
    d = {}
    for _, kind in clothes:
        if kind in d:
            d[kind] += 1
        else:
            d[kind] = 1
    
    for count in d.values():
        answer *= (count + 1)
    
    return answer-1

 

함수의 인자로 의상의 이름과 의상의 종류가 담긴 배열 clothes가 주어졌다.

여기서 뽑아내야 할 것은 의상의 종류와 해당 종류의 의상 개수이다.

 

빈 딕셔너리 d를 선언하고 첫 번째 for문을 돌며 key를 의상의 종류, value를 해당 종류의 의상 개수인 값을 넣어주었다.

d에 해당 key가 있으면 value를 +1하고 없으면 1을 설정해주는 식이다.

 

두 번째 for문에서는 해당하는 경우의 수를 구한다.

 

위의 입력 예제에서 첫 번째 예제의 경우를 살펴보자.

입력으로 [["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]가 주어졌다.

headgear에 해당하는 의상이 yellow_hat, green_turban이고 eyewear에 해당하는 의상이 blue_sunglasses이다.

 

이때 경우의 수를 각각 나눠보면 다음과 같다.

• headgear : yellow_hat 선택, green_turban 선택, 둘 다 선택하지 않음
• eyewear : blue_sunglasses 선택, 선택하지 않음

따라서 (headgear 경우의 수) * (eyewear 경우의 수) - (headgear와 eyewear 둘 다 선택하지 않을 경우의 수) = 3 * 2 - 1 = 5가 된다.

 

그러니 이를 표준화 시켜보면,

(종류 1의 개수 + 종류 1 고르지 않음) * ... * (종류 N의 개수 + 종류 N 고르지 않음) - 모든 종류를 고르지 않음

 = (종류 1의 개수 + 1) * ... * (종류 N의 개수 + 1) - 1 이 된다.

 

결론적으로 두 번째 for문에서 d의 value값만을 뽑아 경우의 수를 구한 뒤 answer-1을 리턴한다.