programmers | Lv2. 위장 [Python]

💡 Python 3

🔗 문제

위장 [Link]

💻 코드

경우의 수를 활용한 문제 주어진 의상 종류의 개수 + 1(입지 않는 경우)로 주어진 의상 종류들의 수를 곱하여 해결

def solution(clothes):
    from collections import defaultdict
    check = defaultdict(int)
    
    for v, k in clothes:
        check[k] += 1
    
    count = 1
    for i in list(check.values()):
        count *= (i + 1)
    
    return  count - 1

💥 다른 사람 코드

Counter와 reduce를 적절하게 활용한 코드
이번에 알고리즘 스터디에서 reduce 알게 되었는데.. 적용을 못해봤다 !!
이번 문제 풀면서 한번더 배웠다 ㅎㅎ

def solution(clothes):
    from collections import Counter
    from functools import reduce
    cnt = Counter([kind for name, kind in clothes])
    answer = reduce(lambda x, y: x*(y+1), cnt.values(), 1) - 1
    return answer