문제
어느 공원 놀이터에는 시소가 하나 설치되어 있습니다. 이 시소는 중심으로부터 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 |
내 풀이
1차
from collections import Counter
def solution(weights):
answer = 0
sit = [2, 3, 4]
for weight in weights:
variables = [weight * i // j for i in sit for j in sit if (weight * i) % j == 0 and i != j ]
for var in variables:
if var in weights:
answer += 1
# 중복 제거
answer /= 2
cnt = Counter(weights)
for i in cnt.values():
if i > 1:
answer += 1
return answer
채점 결과
정확성: 17.6
합계: 17.6 / 100.0
- 몸무게 * (2, 3, 4) / (2, 3, 4)한 값이 같이 탔을때 균형을 이루는 경우
- 따라서 리스트 컴프리헨션을 사용해 만들고 리스트에 있는지 체크하면 되지 않을까 생각 => 시간초과 및 실패
from collections import defaultdict
def solution(weights):
answer = 0
dic = defaultdict(int)
for w in weights:
answer += dic[w] + dic[w*2] + dic[w/2] + dic[(w*2)/3] + dic[(w*3)/2] + dic[(w*4)/3] + dic[(w*3)/4]
dic[w] += 1
return answer
- defaultdict를 생성해서 for문을 하나만 사용
- 어느 경우에 탈 수 있는지 제대로 확인