문제 설명
어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 2(m), 3(m), 4(m) 거리의 지점에 좌석이 하나씩 있습니다.이 시소를 두 명이 마주 보고 탄다고 할 때, 시소가 평형인 상태에서 각각에 의해 시소에 걸리는 토크의 크기가 서로 상쇄되어 완전한 균형을 이룰 수 있다면 그 두 사람을 시소 짝꿍이라고 합니다. 즉, 탑승한 사람의 무게와 시소 축과 좌석 간의 거리의 곱이 양쪽 다 같다면 시소 짝꿍이라고 할 수 있습니다.사람들의 몸무게 목록 weights이 주어질 때, 시소 짝꿍이 몇 쌍 존재하는지 구하여 return 하도록 solution 함수를 완성해주세요.
제한 사항
- 2 ≤
weights의 길이 ≤ 100,000 - 100 ≤
weights[i] ≤ 1,000- 몸무게 단위는 N(뉴턴)으로 주어집니다.
- 몸무게는 모두 정수입니다.
입출력 예
| weights | result |
|---|---|
| [100,180,360,100,270] | 4 |
내 코드
from collections import Counter
def solution(weights):
answer = 0
c = Counter(weights)
for w in weights:
c[w] -=1 # 값 초기화 하면서 시작
answer += c[w] + c[w*2] + c[w/2] + c[(w*2)/3] + c[(w*3)/2] + c[(w*3)/4] + c[(w*4)/3] # y = x*a / b 에 기반하여 모든 비율의 가짓수를 더함
c[w] += 1 # dict에 해당 weight count를 반영
return answer // 2 # 중복해서 값이 들어갈수도 있기 때문에 2로 나눠줌 아이디어
weights 배열의 크기가 이므로 상수시간에 처리할 수 있는 부분이 필요했고, 이를 Counter 모듈을 활용하여 weights 배열의 각 갯수와 dict로 동시에 처리
이때 비율은 2:4 = 1:2 인것들에 대해서만 유의해서 처리
