본문 바로가기
Data Engineering/Python

[코테] 프로그래머스 - 시소 짝꿍 Python 코드

by strongstar 2025. 10. 15.
from collections import Counter

def solution(weights):
    answer = 0
    weights.sort()
    counter = Counter(weights)
    for (w, c) in counter.items():
        answer += c * (c-1) // 2        # 1:1
        answer += c * counter[w*2]      # 2:1
        answer += c * counter[w*3/2]    # 3:2
        answer += c * counter[w*4/3]    # 4:3
    return answer
    
# https://school.programmers.co.kr/learn/courses/30/lessons/152996