[프로그래머스] 의상

yewon Lee·2023년 5월 17일
0


😎코딩테스트 연습>해시>의상


📘 문제풀이

뭔가 코딩이라기보다 수학문제 풀듯이 풀어버린..


종류마다 의상이 n(1이상 다 다른 수)개 씩 있다고 가정할 때 한 종류당 의상을 선택할 수 있는 가지수가 n+1 (+1은 선택하지 않았을 경우)
각각 (n+1)을 곱하고 모든 종류를 선택하지 않았을 경우 -1을 해준다.

def solution(clothes):
    answer = 1
    dicClothes = {}
    
    for i, j in clothes:
        if j in dicClothes:
            dicClothes[j] += 1
        else:
            dicClothes[j] = 1
            
    for v in dicClothes.values():
        answer *= (v+1)
    answer -= 1      
    
    return answer
profile
시작

0개의 댓글