[Level 2] 위장 (Python)

코딩테스트 연습 - 위장

 

def solution(clothes):
    clothes_num = {}
    
    for i in clothes:
        if i[1] in clothes_num:
            clothes_num[i[1]] += 1
        else:
            clothes_num[i[1]] = 1
            
    cnt = 1
    
    for i in clothes_num.values():
        cnt *= i+1
    
    cnt -= 1
    
    return cnt

 

▷ 이 문제는 키와 값으로 이루어진 딕셔너리를 이용하여 풀 수 있다. 의류의 종류를 키로, 해당 종류에 포함된 의류를 값으로 둔다. 첫 번째 for 문은 이를 구현하기 위한 코드이다.

 

▷ 전체 경우의 수는 각 해당 의류의 수에 1을 더한 후, 모두 곱한 뒤, 1을 빼준 것을 통해 구할 수 있다.

 

▶ 딕셔너리는 values 함수를 이용하여 모든 키의 값을 구할 수 있다. 이때, 출력되는 값은 리스트가 아니므로 인덱싱이 필요한 경우 list 함수를 이용하여 형태를 바꾸어 주자.