bit가 눈 앞에서 왔다갔다

Py) 프로그래머스 42578 본문

카테고리 없음

Py) 프로그래머스 42578

헬린인형 2021. 12. 29. 18:36

https://programmers.co.kr/learn/courses/30/lessons/42578

 

코딩테스트 연습 - 위장

 

programmers.co.kr

 

def solution(clothes):
    mul = 1
    answer = 0
    dict = {}
    for i in clothes:
        if i[1] not in dict:
            dict[i[1]] = 1   # 추가
        else:
            dict[i[1]] += 1


    for j in dict:
        mul *= (dict.get(j) + 1)

    answer = mul - 1
    return answer

 

*

딕셔너리 자료형

 

- 특정 종류를 안입을 수도 있다는걸 고려 안했었음

- 이 경우, 아예 안입는 경우는 없으니까 -1도 해줘야함

 

**

(좀 더 깔끔한 걸 보게 돼서 참고용으로)

def solution(clothes):
    answer = 1
    dic = dict()

    for name, kind in clothes:
        if kind not in dic:
            dic[kind] = 1
        else:
            dic[kind] += 1

    for val in dic.values():
        answer *= (val + 1)
    return answer - 1
반응형
Comments