[프로그래머스] 해시 - 위장 (level.2)

 

문제링크

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

코딩테스트 연습 - 위장

 

나의풀이

• (a종류의 옷 개수 x b종류의 옷 개수 x c종류)의 옷 개수 로 구했는데, 이렇게하게 되면abc 모두 입게되는 경우의 수만 계산된다. 따라서 a의 옷을걸치지 않는 옷x의 종류가 한가지 포함되어있다고 생각해야 a는 안입고 b와c만을 입는 경우의 수 까지 계산된다.
  따라서 {(a종류의 옷 개수 + 1) x  (b종류의 옷 개수 + 1)  x  (c종류의 옷 개수 + 1)} 로 계산해야한다. 이것을 생각해내는데 조금 시간이 거렸던것 같다. (원래 경우의수 계산에 좀 약하다ㅠㅠ)
• 마지막에 total -1 을 진행하는 이유는 abc 중에서는 꼭 하나 이상을 걸쳐야 하기 때문에 모두 옷을 걸치지 않은 경우 한가지를 빼준것이다. 

def solution(clothes):

    dic = {}

    for cloth in clothes: #종류별로 dic으로 묶어주기
        if cloth[1] in dic:
            dic[cloth[1]].append(cloth[0])
        else:
            dic[cloth[1]] = [cloth[0]]

    total = 1 # 종류+1 = 옷x경우까지 추가해서 계산
    for d in dic:
        total *= len(dic[d])+1
    return total -1  #모두 옷x인경우 빼줘야함

 

다른사람 풀이

https://programmers.co.kr/learn/courses/30/lessons/42578/solution_groups?language=python3&type=all 

프로그래머스

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

많은사람들이 Counter 을 사용했다. Counter 함수를 활용하는 방법을 익히는 것이 필요하다. 나는 itertools 함수 활용도가 낮다 ㅠㅠ 이 함수 활용도를 높일 수 있도록 추가 공부를 진행해야 겠다.