프로그래머스/위장/py

https://school.programmers.co.kr/learn/courses/30/lessons/42578

프로그래머스

문제

---

스파이가 가진 의상들의 조합의 개수를 구하여라.

 

입력

---

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

출력

---

총 조합의 개수

 

풀이

---

옷의 종류를 각각 a,b,c 라고 할때 a + b + c + (a * b) + (b * c) + (b * a) + (a * b * c) = (1 + a) * (1 + b) * (1 + c) - 1 이다. 따라서 각 옷의 종류를 돌때 값 + 1 을 곱해주고 결과 값에 -1 을 해주면 된다.

 

def solution(clothes):
    answer = 1
    clothDict = {} # dict 로 값 저장
    for i in range(len(clothes)):
        kindOfClothes = clothes[i][1]
        if not kindOfClothes in clothDict:
            clothDict[kindOfClothes] = 1
        else:
            clothDict[kindOfClothes] += 1

    for key,val in clothDict.items(): # 각 dict 값 + 1 을 곱해준다.
        answer *= (val + 1)
    return answer -1  # 마지막 값에 1을 빼준다.