[Programmers] 시소 짝꿍 - 152996

Counter를 이용하여 경우의 수로 나눈 값들 계산하기.

 

Question

 

어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 2(m), 3(m), 4(m) 거리의 지점에 좌석이 하나씩 있습니다.
이 시소를 두 명이 마주 보고 탄다고 할 때, 시소가 평형인 상태에서 각각에 의해 시소에 걸리는 토크의 크기가 서로 상쇄되어 완전한 균형을 이룰 수 있다면 그 두 사람을 시소 짝꿍이라고 합니다. 즉, 탑승한 사람의 무게와 시소 축과 좌석 간의 거리의 곱이 양쪽 다 같다면 시소 짝꿍이라고 할 수 있습니다.
사람들의 몸무게 목록 weights이 주어질 때, 시소 짝꿍이 몇 쌍 존재하는지 구하여 return 하도록 solution 함수를 완성해주세요.

 

Point

 

1. 두 값이 같을 때 ) n명이라면 2명씩 짝지어주는 조합 nC2 = n(n-1)/2 개.
2. 두 값의 비가 2:3일 때 ) 비교 대상을 3 곱하고 2로 나눠서 해당 몸무게가 counter에 있는지 확인.
3. 두 값의 비가 2:4일 때 ) 비교 대상을 2 곱해서 해당 몸무게가 counter에 있는지 확인.
4. 두 값의 비가 3:4일 때 ) 비교 대상을 4 곱하고 3으로 나눠서 해당 몸무게가 counter에 있는지 확인.

 

Code

 

from collections import Counter
def solution(weights):
    answer = 0
    counter = Counter(weights)
    
    for i in range(100, 1001): # 1:1, 2:3, 2:4, 3:4 인 경우의 수.
        if counter[i] > 0:
            answer += (counter[i] * (counter[i]-1))/2 # 1:1 조합 nC2 = n(n-1)/2
            answer += counter[i] * counter[i * 3 / 2] # 2:3
            answer += counter[i] * counter[i * 2] # 2:4
            answer += counter[i] * counter[i * 4 / 3] # 3:4
    
    return answer

 

 

now me

On my github