[프로그래머스] / [Level 2] / [Python] 시소 짝꿍

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

프로그래머스

---

다음과 같이 조합을 사용하면 시간 초과 에러를 출력한다.

from itertools import combinations as c

def solution(weights):
    answer = 0
    weights = sorted(weights)

    ratios = [1/1, 1/2, 2/3, 3/4]
    
    for w in list(c(weights, 2)):
        a, b = w
        
        if a/b in ratios:
            answer+=1

    return answer

현재 탐색하는 몸무게와 시소 거리의 모든 비율을 곱한 값을 dictionary에 저장 후
이후에 같은 (몸무게 * 비율)을 탐색하면 answer에 더하는 방법으로 해결하였다.

from collections import defaultdict

def solution(weights):
    answer = 0

    weights_dict = defaultdict(float)
    ratio = [1/1, 1/2, 2/1, 2/3, 3/2, 3/4, 4/3]
    
    for w in weights:
        for r in ratio:
            answer += weights_dict[r * w]
        weights_dict[w] += 1

    return int(answer)