문제

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

입력

weights : [100,180,360,100,270]

출력

4

제한 사항

2 ≤ weights의 길이 ≤ 100,000
100 ≤ weights[i] ≤ 1,000

• 몸무게 단위는 N(뉴턴)으로 주어집니다.
• 몸무게는 모두 정수입니다.

내가 했던 풀이 방법

나올 수 있는 경우는 다음과 같다. (1, 1), (1, 2), (2, 3), (3, 2) weights를 즉 1, 2, 2/3, 3/2로 곱한 값이 weights에 존재할 경우, 시소 짝꿍이 될 수 있다는 것이다.

1. weights를 내림차순 정렬해준다. (무거운 사람들부터 가벼운 사람 순으로 계산해야 계산에 중복이 없다.)
2. 각각의 weight를 ratio로 곱해준 값이 map에 존재한다면, 해당 value를 answer에 더해준다. 예를 들어 360-2 일 때, 새로 계산한 값이 360이라면 2를 더해주는 것이다. 360이 나오는 weight들과 시소 짝꿍이기 때문이다.
3. map에 value를 1 증가시켜준다. 다음에 계산하는 weight도 360이 나온다면 이번에는 시소짝꿍이 3이 추가가 되는 것이다.

코드

function solution(weights) {
    let answer = 0;
    const map = new Map();
    const ratio = [1, 3 / 2, 2, 4 / 3];
    weights.sort((a, b) => b - a);
    
    weights.forEach(w => {
        ratio.forEach(r => {if (map.has(w * r)) answer += map.get(w * r)});
        map.set(w, (map.get(w) || 0) + 1);
    });

    return answer;
}

회고

weight 별로 나올 수 있는 무게를 모두 찾고 일치하는 경우를 찾도록 했는데 시간초과가 나왔다. 도저히 모르겠어서 찾아보니 경우의 수가 (1, 1), (1, 2), (2, 3), (3, 2)밖에 없다는 걸 알게 되었다. 풀이 자체를 나중에 풀면... 기억할 수 있을까 싶은 문제였다. 개인적으로는 최근 풀었던 문제 중에 제일 어려운 것 같다.