파이썬으로 부시자! Python with Coreenee

프로그래머스 - 해시 - 위장

문제

스크린샷 2020-04-15 오후 5 29 35

입력

스크린샷 2020-04-15 오후 5 29 43


나의 처음 코드

dict로 입력 받은 종류에 숫자를 카운트해서 조합으로 계산해주면 된다. 마지막에 return 에서 -1 해주는 것은 아무것도 입지 않았을 때를 제외해주기 위해서이다.

def solution(clothes):
check_arr = {}
tmp = 1

for i in clothes:
    if i[-1] in check_arr:
        check_arr[i[-1]] += 1
    else:
        check_arr[i[-1]] = 1

#조합 계산 부분
for i in list(check_arr.values()):
    tmp *= (i+1)
        
return tmp - 1

자료구조만 잘 세우면 쉬운 해쉬!