Written by 최태열
on 

[Python][프로그래머스][Level2] 위장

[Python][프로그래머스][Level2] 위장

코드

def solution(clothes):
    fashion=dict()
    for cloth, what in clothes:
        if what in fashion:
            fashion[what].append(cloth)
        else:
            fashion[what]=[cloth]
    answer=1
    for name in fashion:
        answer*=len(fashion[name])+1
    return answer-1

각 부위별 옷 목록을 fashion dictionary를 만들어 저장했다.

같은 what이면 같은 fashion[what] 안에 들어가 있는다.

각 부위를 입을 수도 안 입을 수도 있기 때문에

모든 부위의 개수 만큼 곱해주면 된다.

하지만 다 안 입은 경우는 빼줘야하기 때문에

answer에 1을 빼게 된다.

Discussion and feedback