프로그래머스: 위장

문제

코딩테스트 연습 - 위장

 

문제 풀이

위장 할 수 있는 물건의 가지수에 따라 경우의 수를 구하는 문제이다. Counter를 이용하여 경우의 수를 쉽게 구할 수 있다.

 

코드

from collections import Counter


def solution(clothes):
    answer = 1
    kinds = Counter([kind for _, kind in clothes])
    nums = [x for x in kinds.values()]
    for num in nums:
        answer *= num + 1
    return answer - 1

• 인자로 주어지는 clothes 에서 종류를 Counter를 사용하여 카운트 한다.
• 물건의 종류에 따라 (종류1 + 1) * (종류2 + 1) ... (종류N + 1) - 1 과 같은 방식으로 위장할 수 있는 조합의 개수를 구할 수 있다.
  • Counter의 반환 값은 dictionary이다.