[ 2023-05-24 🐟 TIL ]

LV.2 시소 짝꿍

문제링크

from collections import Counter


def solution(weights):
    answer = 0
    w_counter = Counter(weights)
    
    # 예제에 값처럼 동일 한 값이 2개 이상일 때 계산할 코드 👇
    for w, c in w_counter.items():
        if c >= 2:  # 같은 무게로 2개 이상이면 nC2으로 계산
            answer += (c * (c - 1)) // 2
    weights = set(weights)  # 같은 무게를 가진 (중복)제거
    
    # 서로 다른 무게가 2/3, 2/4, 3/4 비율을 가지고 있을 때를 찾는 코드 👇
    for w in weights:
      # w무게에 2/3, 2/4, 3/4에 해당하는 w의 갯수를 가지고 계산 가능
        if (w * 2/3) in weights:
            answer += w_counter[w*2/3] * w_counter[w]
            # w무게를 가진 갯수와 w*2/3값을 가진 갯수의 곱
            # ex) [180, 270, 100, 270] 인 배열일 때
            # 270의 2/3인 180이 2개 있다. 
            # 270도 2개 있으니 2 * 2 된다.
        if (w * 2/4) in weights:
            answer += w_counter[w*2/4] * w_counter[w]
        if (w * 3/4) in weights:
            answer += w_counter[w*3/4] * w_counter[w]

    return answer

해당 문제는 검색을 통해서 해결한 문제입니다.
아직 저도 이해가 85%정도..? 풀이가 틀렸으면 알려주세요..🧎🏻‍♂️