[프로그래머스] 시소짝꿍 (Python)

link 🔗 https://school.programmers.co.kr/learn/courses/30/lessons/152996

프로그래머스

 

🔥 dict에 weight의 개수를 모두 기록하는 방식까진 떠올렸는데, 그 후로 잘 풀리지 않아서 다른 분의 풀이를 참고해서 풀었다.

🔥 나올 수 있는 경우의 수는 (1:1), (2:3), (2:4), (3:2), (3:4), (4:2), (4:3) 으로 7가지이다.

from collections import defaultdict

def solution(weights):
    answer = 0
    dict = defaultdict(int)

    for wei in weights:
        answer += dict[wei]+dict[wei*(2/3)]+dict[wei*(2/4)]+dict[wei*(3/2)]+dict[wei*(3/4)]+dict[wei*2]+dict[wei*(4/3)]
        dict[wei] += 1
        
    return answer

 

defaultdict

- 파이썬의 내장 모듈인 collections의 클래스

- 딕셔너리의 기본값 설정 가능

from collections import defaultdict

# defaultdict(<class 'int'>, {})
dict_int = defaultdict(int)
# 0
dict_int['test']

# defaultdict(<class 'list'>, {})
dict_list = defaultdict(list)
# []
dict_list['test']

# defaultdict(<class 'set'>, {})
dict_set = defaultdict(set)
# set()
dict_set['test']