[Programmers] 위장

 

 

코드

 

def solution(arr):
    dict = {}
 
    for i in range(len(arr)):
        if arr[i][1] not in dict:
            dict[arr[i][1]] = [arr[i][0]]
        else:
            dict[arr[i][1]].append(arr[i][0])
    
    k = list(dict.values())
    cnt = 1
    for i in k:
        cnt *= len(i) + 1
    return cnt - 1

분석 > 해시맵을 만드는 것까지는 했는데, 조합 수를 구하는 것에서 막혔던 문제였다. 그래서 구글링으로 아이디어를 얻었다. 예를들어, 배열[[2,3,4],[1,0]] 이 있다면 첫번째 배열 내에서 나올 수 있는 조합의 수는 < 2를 사용, 3을 사용, 4를 사용, 아무것도 사용하지 않음 > 이렇게 총 4가지 조합이 나오고, 두 번째 배열에서는 <1을 사용, 0을 사용, 아무 것도 사용하지 않음> 이렇게 총 3가지의 조합이 나온다. 그리하여 총 조합의 수는 3*4로 12가지가 나오고, 아무것도 입지 않는 경우를 제외하면 총 11가지 조합을 구할 수 있다는 아이디어였다. 그래서 이를 추가해 코드를 완성하니 쉽게 풀렸다. 

 

 

 

코딩테스트 연습 - 위장