Programmers / 2단계 / 시소 짝꿍 / python

 

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

 

코드

from collections import Counter
def solution(weights):
    answer = 0
    times = (3/4, 2/3, 1/2)
    count_dict = Counter(weights)
    for key, value in count_dict.items():
        if value > 1:
            answer += value * (value - 1) / 2
    
    # weights 안에 중복 없애기 
    weights = list(set(weights))
    
    # 같은 무게가 아닐 경우
    for i in weights:    
        for j in times:
            if i * j in weights:
                combination_num = count_dict[i * j] # 조합 갯수
                answer += combination_num * count_dict[i]
            
    return answer

 

 

참조

https://jibinary.tistory.com/48

[Python] [Level 2] 시소 짝꿍 (Counter 함수 활용하기)