프로그래머스 위장 (python)

코딩테스트 연습 - 위장

코딩테스트 연습 - 위장

해시를 이용하는 문제였기 때문에 딕셔너리를 사용해서 문제를 풀어야겠다고 생각했다. 아직 파이썬의 딕셔너리를 잘 활용할 줄 몰라서 필기해두었던 노트를 참고하여 문제를 해결했다. 딕셔너리에서 for문을 이용해 값을 빼오는 법도 배울 수 있었다.

#key들 만 출력
print(cabinet.keys())
#value만 출력
print(cabinet.values())
# key-value 쌍 출력
print(cabinet.items())

 

풀이

def solution(clothes):
    lst = {}
    for cloth in clothes:
        if cloth[1] in lst: lst[cloth[1]]+=1
        else:
            lst[cloth[1]]=1

    cnt = 1
    for i in lst.values():
        cnt *=(i+1)
    return cnt-1